GridSearchCV:将权重传递给得分手

时间:2016-05-02 09:48:19

标签: python scikit-learn cross-validation grid-search

我正在尝试使用GridSearchCV为XGB_Classifier找到最佳参数集。 由于我的数据非常不平衡,因此必须使用权重来执行拟合和评分(在交叉验证中),因此我必须使用自定义记分器,该记分器采用“权重”和“权重”。矢量作为参数。 但是,我找不到让GridSearchCV通过'权重'矢量给一个得分手。

尝试将此功能添加到gridsearch:

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

但他们没有合并为主人,现在我担心这段代码与上游变化不兼容。

有没有人遇到类似的问题,并且有任何“容易”的问题。应对它的方式?

1 个答案:

答案 0 :(得分:1)

您可以在Scikit-learn balanced subsampling

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