1000字范文,内容丰富有趣,学习的好帮手!
1000字范文 > sklearn波士顿房价数据集——线性回归

sklearn波士顿房价数据集——线性回归

时间:2019-11-09 01:46:34

相关推荐

sklearn波士顿房价数据集——线性回归

(一)导入并查看数据

import pandas as pd import matplotlib.pyplot as pltfrom sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression #线性回归库from sklearn.datasets import load_boston #导入波士顿数据集import matplotlib.pyplot as pltimport numpy as npimport pandas as pd

df = load_boston()x = df.data #数据y = df.target #标签

x.shape

x_train,x_test,y_train,y_test = train_test_split(x,y,test_size = 0.2,random_state = 888)

(random_state为随机数种子)

(二)训练

train = LinearRegression()train.fit(x_train,y_train)

train.score(x_test,y_test) #查看准确率

train.coef_ #每一列数据斜率,对应x的系数,13位未知数

(三)预测

train.predict(x_test)

(四)评估

dev = train.predict(x_test) - y_test #偏差dev

均方根误差

RMSE = np.sum(np.sqrt(dev**2))/102RMSE

(五)保存结果

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。