2

问题

我想在进行多标签分类时使用验证数据集提前停止，但似乎 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参数没有表明现在需要在多标签数据集的训练期间评估模型。这不支持吗？还是我做错了什么？

4

1 回答 1

1

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

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

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

（来源）到

        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]))
于 2021-06-08T18:26:07.243 回答