scikit-learn的.predict()默认阈值
- 论坛
- scikit-learn的.predict()默认阈值
19 浏览
scikit-learn的.predict()默认阈值
我正在处理一个类别不平衡的分类问题(1的比例为5%)。我想要预测类别,而不是概率。
在一个二分类问题中,scikit的classifier.predict()
默认使用0.5
作为阈值吗?
如果不是,默认的方法是什么?如果是,我该如何改变它?
在scikit中,一些分类器有class_weight='auto'
选项,但并非所有分类器都有。使用class_weight='auto'
,.predict()
会使用实际人口比例作为阈值吗?
在像MultinomialNB
这样不支持class_weight
的分类器中,有什么其他方法可以做到这一点?除了使用predict_proba()
然后自己计算类别。