python scikit自定义得分手

Posted 2021-05-09

## scorer
from sklearn.metrics import make_scorer
def RMSE(y_true,y_pred):    
    error = y_pred - y_true
    mse = np.mean(error**2)
    return np.sqrt(mse)
def scorer():
    return make_scorer(RMSE, greater_is_better=False)