- 54
- 0
- 约4.52千字
- 约 4页
- 2021-02-14 发布于山西
- 举报
六六 DATA
模型评估-分类与回归(python 之sklearn 实现)
X_train = []
X_test = []
y_train = []
y_test = []
y_hat = []
#%%######回归模型评估######
import numpy as np
from sklearn.linear_model import LinearRegression
lr = LinearRegression()
lr.fit(X_train,y_train)
from sklearn.metrics import mean_squared_error, mean_absolute_error, r2
_score
print(均方误差(MSE):,mean_squared_error(y_test,y_hat))
print(根均方误差(RMSE):,np.sqrt(mean_squared_error(y_test,y_hat)))
print(平均绝对误差(MAE):,mean_absolute_error(y_test,y_hat))
print(训练集R^2:,r2_score(y_train,lr.predict(X_train)))
原创力文档

文档评论(0)