我知道我可以像这样实现一个均方根误差函数:
def rmse(predictions, targets):
return np.sqrt(((predictions - targets) ** 2).mean())
如果这个 rmse 函数是在某个库中实现的，我在寻找什么，可能是在 scipy 或 scikit-learn 中？
最佳答案
sklearn >= 0.22.0
sklearn.metrics 有一个 mean_squared_error 函数和一个 squared kwarg(默认为 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))
关于python - python中是否有用于均方根误差(RMSE)的库函数？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/17197492/