點選檢視程式碼
import numpy as np
import statsmodels.api as sm
import pylab as plt
a = np.loadtxt('F:\python數學建模與演算法\源程式\《Python數學建模演算法與應用》程式和資料//10第10章 迴歸分析/data10_2.txt')
plt.rc('font', family='SimHei')
plt.rc('axes', unicode_minus=False)
plt.plot(a[0], a[2], '*', label='$x_1$')
plt.plot(a[1], a[2], 'o', label='$x_2$')
plt.legend(loc='upper left')
d = {'x1': a[0], 'x2': a[1], 'y': a[2]}
re = sm.formula.ols('y~x1+x2', d).fit()
print(re.summary())
yh = re.predict({'x1': [9, 10], 'x2': [10, 9]})
print('殘差的方差:', re.mse_resid)
print('預測值：', yh); plt.show()
