使用 SKlearn 进行多标签分类 - 如何使用验证集？

Question

问题

我想在进行多标签分类时使用验证数据集来提前停止，但 sklearn 的 MultiOutputClassifier 似乎不支持它。您对解决方案有什么建议吗？

我做了什么

import numpy, sklearn
from sklearn.multioutput import MultiOutputClassifier
from xgboost import XGBClassifier

# Creating some multi-label data
X_train = numpy.array([[1,2,3],[4,5,6],[7,8,9]])
X_valid = numpy.array([[2,3,7],[3,4,9],[7,8,7]])
Y_train = numpy.array([[1,0],[0,1],[1,1]])
Y_valid = numpy.array([[0,1],[1,1],[0,0]])

# Creating a multi-label xgboost
xgb = XGBClassifier(n_estimators=500, random_state=0, learning_rate=0.05, eval_metric='logloss')
xgb_ml = MultiOutputClassifier(xgb)

# Training the model
xgb_ml.fit(X_train, Y_train)

到这里为止一切正常！

现在我想使用验证集来做一些提前停止。我使用与普通单标签 xgboost 相同的参数。

# Training model using an evaluation dataset
xgb_ml.fit(X_train, Y_train, eval_set=[(X_train, Y_train), (X_valid, Y_valid)], early_stopping_rounds=5)
>ValueError: y should be a 1d array, got an array of shape (3, 2) instead.

似乎 eval_set 参数没有发现现在需要在多标签数据集上训练期间评估模型。这不支持吗？还是我做错了什么？

Answer 1

@afsharov 在评论中指出了这个问题。 sklearn 对 fit_params 一无所知，它只是将它们传递给各个单输出模型。

MultiOutputClassifier 做的不多，所以简单地遍历目标、拟合 xgboost 模型并将它们保存到列表中并不是什么大问题。主要的打击似乎是并行化的损失，但你也可以自己做。

如果你真的想要所有东西都包含在class中，我认为从MultiOutputClassifier派生并覆盖fit方法应该是足够。您将复制大部分原始 fit 方法（classes_ 属性设置和大多数父 class _MultiOutputEstimator 的 fit 方法），但会破坏 eval_set 将第二个元素放入它们的列中并将它们压缩在一起以进行平行装配。大致如下：

# current code
        fit_params_validated = _check_fit_params(X, fit_params)

        self.estimators_ = Parallel(n_jobs=self.n_jobs)(
            delayed(_fit_estimator)(
                self.estimator, X, y[:, i], sample_weight,
                **fit_params_validated)
            for i in range(y.shape[1]))

(source) 至

        fit_params_validated = _check_fit_params(X, fit_params)
        eval_set = fit_params_validated.pop("eval_set", [(X, y)])
        eval_set_sliced = [(eval_set_i[0], eval_set_i[1][:, i]) for eval_set_i in eval_set]

        self.estimators_ = Parallel(n_jobs=self.n_jobs)(
            delayed(_fit_estimator)(
                self.estimator, X, y[:, i], sample_weight,
                eval_set=eval_set_sliced[i],
                **fit_params_validated)
            for i in range(y.shape[1]))

使用 SKlearn 进行多标签分类 - 如何使用验证集？

Multi-label classification with SKlearn - How do you use a validation set?

python

scikit-learn

xgboost

multilabel-classification

early-stopping

问题

我做了什么