单变量线性回归-03 数据预测
数据集一般划分为: 训练集(TrainDataSet),验证集(Validation DataSet), 测试集(Test DataSet)
可以使用sklearn中的cross_validaton程序包划分。
下面进行预测
from sklearn.linear_model import LinearRegression
lr = LinearRegression()
lr.fit(X_train,y_train)
print('截距: %f' % lr.intercept_)
print('系数: %f' % lr.coef_)
y_pred = lr.predict(X_test)

浙公网安备 33010602011771号