def make_scoring( scoring):
if scoring == 'r2':
return make_scorer( metrics.r2_score)
elif scoring == 'mean_absolute_error':
return make_scorer( metrics.mean_absolute_error, greater_is_better=False)
elif scoring == 'mean_squared_error':
return make_scorer( metrics.mean_squared_error, greater_is_better=False)
elif scoring == 'median_absolute_error':
return make_scorer( metrics.median_absolute_error, greater_is_better=False)
else:
raise ValueError("Not supported scoring")
评论列表
文章目录