scikit-learn的.predict()默认阈值

19 浏览
0 Comments

scikit-learn的.predict()默认阈值

我正在处理一个类别不平衡的分类问题(1的比例为5%)。我想要预测类别,而不是概率

在一个二分类问题中,scikit的classifier.predict()默认使用0.5作为阈值吗?

如果不是,默认的方法是什么?如果是,我该如何改变它?

在scikit中,一些分类器有class_weight='auto'选项,但并非所有分类器都有。使用class_weight='auto'.predict()会使用实际人口比例作为阈值吗?

在像MultinomialNB这样不支持class_weight的分类器中,有什么其他方法可以做到这一点?除了使用predict_proba()然后自己计算类别。

0