linear_model = LinearRegression() linear_model.fit(X_train, y_train) coef = linear_model.coef_#回归系数 line_pre = linear_model.predict(X_test) print('SCORE:{:.4f}'.format(linear_model.score(X_test, y_test))) print('RMSE:{:.4f}'.format(np.sqrt(mean_squared_error(y_test, line_pre))))
时间: 2023-04-05 21:04:55 浏览: 198
LinearRegression-model:这是用Python编码的线性回归模型,适用于用于处理2D数据集的普通最小二乘法
linear_model = LinearRegression() 表示创建一个线性回归模型。
linear_model.fit(X_train, y_train) 表示用训练数据 X_train 和 y_train 训练模型。
coef = linear_model.coef_ 表示获取模型的系数。
阅读全文