python中是否有均方根误差（RMSE）的库函数？

Question

我知道我可以像这样实现均方根误差函数：

 def rmse(predictions, targets):
    return np.sqrt(((predictions - targets) ** 2).mean())

如果这个 rmse 函数是在某处的库中实现的，我正在寻找什么，也许是在 scipy 或 scikit-learn 中？

原文由 siamii 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

sklearn >= 0.22.0

sklearn.metrics has a mean_squared_error function with a squared kwarg (defaults to True ).将 squared 设置为 False 将返回 RMSE。

 from sklearn.metrics import mean_squared_error

rms = mean_squared_error(y_actual, y_predicted, squared=False)

sklearn < 0.22.0

sklearn.metrics 有一个 mean_squared_error 功能。 RMSE 只是它返回的任何值的平方根。

 from sklearn.metrics import mean_squared_error
from math import sqrt

rms = sqrt(mean_squared_error(y_actual, y_predicted))

原文由 Greg 发布，翻译遵循 CC BY-SA 4.0 许可协议