sklearn使用——最小二乘法

参考网页：http://sklearn.apachecn.org/cn/0.19.0/

其中提供了中文版的文件说明，较为清晰。

 from sklearn.linear_model import LinearRegression as lr
 import matplotlib.pyplot as plt
 import numpy as np

 x = np.array([3.6,4.5,2.6,4.9,2.5,3.5]).reshape(-1,1)
 y = np.array([9.7,8.1,7.6,8.6,9.0,7.8]).reshape(-1,1)

 model = lr()

 model.fit(x,y)

 y_plot = model.predict(x)

 print(model.coef_)

 plt.scatter(x,y,color='red',label= "sample data",linewidth=2)
 plt.plot(x,y_plot,color='green',label="fitting line",linewidth=2)
 plt.legend(loc='lower right')
 plt.show()

（数据随便编的，不标准）

最小二乘法的实现利用了sklearn中提供的linear_model线性模型，numpy提供对数组的操作，matplotlib.pyplot提供对模型的图例化。

其中：

reshape(-1,1)利用numpy提供的方法将数组转换一维数组。

model实例化LinearRegression方法。

fit（x，y）为监督学习算法，若改为fit（X，y=None）则为无监督学习算法。

plt.scatter为散点图方法，此处用于将各个点标于图上。

plt.plot为折线图方法，此处用于将预测得的拟合直线标于图上。

plt.legend为图例设置方法，用于标记图例的显示位置，可用参数主要为“lower，upper，center，right，left，best”（若留空则为右上角，与best相同）。

plt.show为图例输出方法。