使用cross_val_score通过交叉验证计算均方误差的函数

Question

I would like to write a function that allows me to calculate the root mean square error obtained by 5-sample cross-validation using the cross_val_score function of sklearn.model_selection.

(Knowing that the scoring argument of the cross_val_score()function allows to choose the metric we want to use.)

I found this method, but it does not correspond to the question :

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

Thank you very much, Merci beaucoup :)

Answer 1

Your using the wrong formula in your code, here is the correct formula for mean square error.

Y is the expected output, O is actual output from neural network.

Answer 2

You can simply set scoring='mean_squared_error' in sklearn.model_selection.cross_val_score . Check out the documentation for the validator and the metric .

In other words:

cv = cross_val_score(estimator=my_estimator, X, y, cv=5, scoring='mean_squared_error')

Answer 3

You can try :

def rmse_cv(model):     
    rmse= np.sqrt(-cross_val_score(model, X, y, scoring="neg_mean_squared_error", cv=5))     
    return rmse