mse = mean_squared_error(y_test, y_pred) rmse = np.sqrt(mse)
时间: 2024-05-17 10:13:06 浏览: 191
MSE.rar_mean_mean-squared error _mse
这是一个使用均方误差(Mean Squared Error,MSE)和均方根误差(Root Mean Squared Error,RMSE)评估回归模型性能的代码片段。其中,y_test是测试集的真实值,y_pred是模型预测的值。均方误差是真实值与预测值之间差的平方和的平均值,即:
$$MSE = \frac{1}{n}\sum_{i=1}^{n}(y_{test,i} - y_{pred,i})^2$$
均方根误差是均方误差的平方根,即:
$$RMSE = \sqrt{MSE}$$
其中,n为样本数量。这两个指标都用于评估回归模型的性能,其值越小表示模型的预测结果与真实值之间的误差越小,模型的性能越好。
阅读全文