线性回归(sklearn 多元线性回归)- 05

使用sklearn进行多元线性回归代码编写

上面,还是一样使用numpy.genfromtxt方法将这个csv文件中间的数据进行加载

这里将上面载入的数据进行数据切分,即分出x0,x1和y值,之后就能够开始实例化 linear_model.LinearRegression()这个方法了

在前面已经使用了model.fit将数据放入到模型中间,现在即可使用model.coef_获取到对应的系数(这里注意,如果前面传入的x_data是两个,这里对应的系数也就是两个,如果传入的x_data是n个,这个系数也就是对应的n个),model.intercept_对应的就是截距

这里看到,创建了一个x_test进行数据的预测,这里的x_test传入的相当于是x0和x1,需要预测y值,这里预测这个值使用的是predict方法

最后就是进行画图了

如上,画图的方式就是上面这样固定的方式进行

猜你喜欢

转载自blog.csdn.net/Escid/article/details/90399310