def cross_validate(estimator, training_data, training_targets):
mse = cross_val_score(estimator, X=training_data, y=training_targets, scoring=root_mean_log_squared_error)
r2 = cross_val_score(estimator, X=training_data, y=training_targets, scoring='r2')
return (-1 * np.mean(mse), np.mean(r2))
评论列表
文章目录