scikit-learn-hello world

作者:管理员 发布时间:2020-12-22 09:32

scikit-learn内置了一些数据集,以及一些常见的机器学习算法

今天,我们来通过几行代码,实现波士顿房价预测!

我们使用内置的数据集,使用线性回归算法,来进行数据训练,通过此模型,进行数据预测;

代码如下:

#拿sklearn内置数据集
from sklearn import datasets
#调用线性回归函数
from sklearn.linear_model import LinearRegression

 
#导入数据集
loaded_data = datasets.load_boston()
#这是训练数据
data_X = loaded_data.data
#这是测试数据
data_y = loaded_data.target
 
#设置线性回归模块
model = LinearRegression()
#训练数据,得出参数
model.fit(data_X, data_y)
 
#取一个数据,进行预测,打印预测结果
jieguoX = model.predict(data_X[:4,:])
print(jieguoX)

#打印测试数据级中的真实结果
JieguoY = data_y[:4]
print(JieguoY)

#自己比较下两组数据,接近吗

在今后的学习中,我们还会用到这个数据集,以及更多的算法;

我们日后再来实现一遍房价预测,看看是否能够准确。


标签:
Copyright © 2020 万物律动 旗下 AI算法狮 京ICP备20010037号-1
本站内容来源于网络开放内容的收集整理,并且仅供学习交流使用;
如有侵权,请联系删除相关内容;