1

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

有一些尝试将此功能添加到网格搜索：

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

但是它们没有合并到 master 中，现在恐怕此代码与上游更改不兼容。

有没有人遇到过类似的问题，有什么“简单”的方法可以解决吗？

4

1 回答 1

1

您可以手动平衡您的训练数据集，如Scikit-learn 平衡子采样的答案

于 2016-05-03T00:36:27.817 回答