Scikit学习上的学习曲线功能

Question

我想使用RMSE作为“评分参数”，为我的回归问题绘制学习曲线。

如何设置其他“得分”参数？似乎在学习曲线函数中不赞成使用“评分” ...

有人可以帮助我吗？或向我推荐其他功能...

Answer 1

在最新版本https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.learning_curve.html中，有scoring个参数。

如果您是从https://scikit-learn.org/stable/auto_examples/model_selection/plot_learning_curve.html#sphx-glr-auto-examples-model-selection-plot-learning-curve-py开始plot_learning_curve功能

您可以这样更改它：

def plot_learning_curve(estimator, title, X, y, axes=None, ylim=None, cv=None,
                        n_jobs=None, train_sizes=np.linspace(.1, 1.0, 5), scoring='metric'):
...

    train_sizes, train_scores, test_scores, fit_times, _ = \
        learning_curve(estimator, X, y, cv=cv, n_jobs=n_jobs,
                       train_sizes=train_sizes,
                       return_times=True, scoring=scoring)
...