GridSearchCV:将权重传递给记分员

GridSearchCV: passing weights to a scorer

我正在尝试使用 GridSearchCV 为 XGB_Classifier 找到最佳参数集。 由于我的数据非常不平衡,拟合和评分(在 cross_validation 中)都必须使用权重执行,因此我必须使用自定义评分器,它将 'weights' 向量作为参数。 但是,我找不到让 GridSearchCV 将 'weights' 向量传递给记分员的方法。

有人尝试将此功能添加到 gridsearch:

https://github.com/ndawe/scikit-learn/commit/3da7fb708e67dd27d7ef26b40d29447b7dc565d7

但他们没有合并到 master 中,现在我担心这段代码与上游更改不兼容。

有没有人遇到过类似的问题,有什么'easy'方法可以解决吗?

您可以按照 Scikit-learn balanced subsampling

的答案手动平衡您的训练数据集