分类问题的评价指标是准确率,那么回归算法的评价指标就是MSE,RMSE,MAE、R-Squared。下面一一介绍 1、MSE(Mean Squared Error)均方误差这里的y是测试集上的。 用 真实值-预测值 然后平方之后求和平均。 猛着看一下这个公式是不是觉得眼熟,这不就是线性回归的损失函数嘛!!! 对,在线性回归的时候我们的目的就是让这个损失函数最小。那么模型做出来了,我们把损失函数丢到测试集上去看看损失值不就好了嘛。简单直观暴力! y_preditc=reg.predict(x_test) #reg是训练好的模型 mse_test=np.sum((y_preditc-y_test)**2)/len(y_test) #跟数学公式一样的
2、RMSE(Root Mean Squared Error)均方根误差这不就是MSE开个根号么。有意义么?其实实质是一样的。只不过用于数据更好的描述。 rmse_test=mse_test ** 0.5
3、MAE (Mean absolute Error)平均绝对误差mae_test=np.sum(np.absolute(y_preditc-y_test))/len(y_test)
4、R-Squared上面的几种衡量标准针对不同的模型会有不同的值。比如说预测房价 那么误差单位就是万元。数子可能是3,4,5之类的。那么预测身高就可能是0.1,0.6之类的。没有什么可读性,到底多少才算好呢?不知道,那要根据模型的应用场景来。 1- mean_squared_error(y_test,y_preditc)/ np.var(y_test)
Scikit-learn中的各种衡量指标 from sklearn.metrics import mean_squared_error #均方误差 from sklearn.metrics import mean_absolute_error #平方绝对误差 from sklearn.metrics import r2_score#R square #调用 mean_squared_error(y_test,y_predict) mean_absolute_error(y_test,y_predict) r2_score(y_test,y_predict)
参考文献: 【1】回归评价指标MSE、RMSE、MAE、R-Squared 【2】回归模型的几个评价指标 来源:http://www./content-4-119751.html |
|