點選檢視程式碼
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')
X = sm.add_constant(a[:2].T)
re = sm.OLS(a[2], X).fit()
print(re.summary())
yh = re.predict(np.array([[1,9,10],[1,10,9]]))
print('殘差的方差:', re.mse_resid)
print('預測值:', yh); plt.show()