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

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

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

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

有没有人遇到过类似的问题,有没有“简单”的方法来解决?

最佳答案

您可以按照Scikit-learn balanced subsampling的答案手动平衡训练数据集

关于python - GridSearchCV:将权重传递给计分员,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/36979490/

10-12 18:57