调整 sklearn.ensemble.GradientBoostingClassifier 中预测方法阈值的方法
Posted
技术标签:
【中文标题】调整 sklearn.ensemble.GradientBoostingClassifier 中预测方法阈值的方法【英文标题】：the way to tune the threshold for the predict method in sklearn.ensemble.GradientBoostingClassifier 【发布时间】：2019-03-18 14:18:29 【问题描述】：我正在使用 scikit-learn (sklearn.ensemble.GradientBoostingClassifier) 中实现的梯度提升分类器来解决二进制分类问题。虽然默认提供了 predict 方法，但官方文档中没有显示任何线索：
默认阈值是如何定义的，
在调优模型期间是否有自定义方法来修改此阈值
欢迎任何线索和建议
非常感谢
【问题讨论】：
【参考方案1】：至于1，我认为是0.5
至于2，当我使用sklearn库时，我使用了predict_proba，然后你可以自己阈值http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingClassifier.html#sklearn.ensemble.GradientBoostingClassifier.predict_proba （这样你也可以通过比较来检查默认阈值是多少）
【讨论】：
以上是关于调整 sklearn.ensemble.GradientBoostingClassifier 中预测方法阈值的方法的主要内容，如果未能解决你的问题，请参考以下文章