相关疑难解决方法(0)

具有评分函数和改装参数的 GridSearchCV

我的问题似乎与类似,但没有可靠的答案。

我正在进行多类多标签分类,为此我定义了自己的评分器。然而,为了refit最终获得参数并获得模型的最佳参数,我们需要引入评分器函数之一进行改装。如果我这样做,我会收到错误消息missing 1 required positional argument: 'y_pred'。y_pred 应该是拟合的结果。但不确定这个问题从何而来以及如何解决。

下面是代码:

scoring = {'roc_auc_score':make_scorer(roc_auc_score),
          'precision_score':make_scorer(precision_score, average='samples'),
          'recall_score':make_scorer(recall_score, average='samples')}

params = {'estimator__n_estimators': [500,800],
          'estimator__max_depth': [10,50],}

model = xgb.XGBClassifier(n_jobs=4)
model = MultiOutputClassifier(model)

cls = GridSearchCV(model, params, cv=3, refit=make_scorer(roc_auc_score), scoring = scoring, verbose=3, n_jobs= -1)

model = cls.fit(x_train_ups, y_train_ups)
print(model.best_params_)
Run Code Online (Sandbox Code Playgroud)

python scikit-learn xgboost gridsearchcv

1
推荐指数
1
解决办法
2463
查看次数

标签 统计

gridsearchcv ×1

python ×1

scikit-learn ×1

xgboost ×1