使用 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 模型并将它们保存到列表中并不是什么大问题。主要打击似乎是并行化的损失，但您也可以自己做。

如果您真的希望所有内容都包含在一个类中，我认为从MultiOutputClassifier 派生并覆盖fit 方法就足够了。您将复制大部分原始 fit 方法（classes_ 属性设置和大部分父类 _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]))