为sklearn.ensemble.GradientBoostingClassifier中的预测方法调整阈值的方法

时间:2018-10-13 16:24:55

标签: python scikit-learn classification

我正在使用在scikit-learn(sklearn.ensemble.GradientBoostingClassifier)中实现的梯度提升分类器来解决二进制分类问题。尽管默认情况下提供了预报方法,但官方文件中没有任何线索:

  1. 如何定义默认阈值,

  2. 在调整模型期间是否有定制的方法来修改此阈值

任何线索和建议都会受到欢迎

非常感谢

1 个答案:

答案 0 :(得分:0)

至于1,我认为是.5

关于2,当我使用sklearn库时,我使用了predict_proba,然后您可以自己设定阈值
http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingClassifier.html#sklearn.ensemble.GradientBoostingClassifier.predict_proba
(通过这种方式,您还可以通过比较来检查默认阈值)