回归评估指标:MSE、R2

{y_{i}}:原数据标签

\hat{y_{i}}:预测结果

\bar{y_{i}}:平均值

1、均方误差:MSE(Mean Squared Error):

2、均方根误差:RMSE

对MSE开平方

3、R2(R-Square):

注:R2一般取(0,1),0表示拟合效果不好。如果出现负值,首先考虑数据集是否有问题,如果是集成模型,考虑若学习模型是否不足。

代码实现:

from sklearn.metrics import mean_squared_error  # 均方误差
from sklearn.metrics import r2_score  # R square

print("均方误差 MSE= ", mean_squared_error(y_test, y_pred))  # 均方误差MSE
print("均方根误差 RMSE= ", sqrt(mean_squared_error(y_test, y_pred)))  # 均方根误差 RMSE
print("R^2= ", r2_score(y_test, y_pred))  # r2

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
好的,下面是实现的代码。 首先,我们需要导入所需的库和数据集: ```python import numpy as np from sklearn.datasets import load_boston from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score # 加载波士顿房价数据集 boston = load_boston() X = boston.data y = boston.target # 将数据集划分为训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) ``` 然后,我们使用线性回归模型对训练数据进行拟合,并预测测试数据的房价: ```python # 使用线性回归模型对训练数据进行拟合 lr = LinearRegression() lr.fit(X_train, y_train) # 预测测试数据的房价 y_pred = lr.predict(X_test) ``` 最后,我们使用三个评估指标MSE,MAE和R2)来评估模型的性能: ```python # 计算MSE(均方误差) mse = mean_squared_error(y_test, y_pred) print("MSE: ", mse) # 计算MAE(平均绝对误差) mae = mean_absolute_error(y_test, y_pred) print("MAE: ", mae) # 计算R2(决定系数) r2 = r2_score(y_test, y_pred) print("R2: ", r2) ``` 完整代码如下: ```python import numpy as np from sklearn.datasets import load_boston from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score # 加载波士顿房价数据集 boston = load_boston() X = boston.data y = boston.target # 将数据集划分为训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) # 使用线性回归模型对训练数据进行拟合 lr = LinearRegression() lr.fit(X_train, y_train) # 预测测试数据的房价 y_pred = lr.predict(X_test) # 计算MSE(均方误差) mse = mean_squared_error(y_test, y_pred) print("MSE: ", mse) # 计算MAE(平均绝对误差) mae = mean_absolute_error(y_test, y_pred) print("MAE: ", mae) # 计算R2(决定系数) r2 = r2_score(y_test, y_pred) print("R2: ", r2) ``` 输出结果如下: ``` MSE: 24.29111947497371 MAE: 3.189091965887853 R2: 0.6687594935356313 ``` 这意味着我们的模型的平均预测误差约为3.19,R2评分为0.67,这表明模型对数据的解释力还可以,但仍有改进的空间。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值