sklearn里有没有直接计算rmse的函数？
1个回答
sklearn里没有计算RMSE的，但是有计算MSE的。你把MSE开根就可以得到RMSE了。
from sklearn.metrics import mean_squared_error
import numpy as np
def rmse(y_true, y_pred):
return np.sqrt(mean_squared_error(y_true, y_pred))
xiaosu
2019-04-03 10:17
相关讨论
随便看看