【发布时间】:2022-01-22 22:23:16
【问题描述】:
我想执行概率二元分类 (0,1)。我的数据集不平衡。因此,我使用 SVC 并为每个类分配了一些类权重。
将SVC拟合到测试数据集后,我使用predict_proba得到概率分类结果。然而,SVC 将训练样例预测为 1,概率分类结果高于 0.4。
我认为predict_proba 的默认阈值为 0.5。
不知道如果使用class_weights,默认阈值会自动变化吗?
例如:
[0.58497606, 0.41502394] >> predict_probaba 函数结果的预测标签为 1。
【问题讨论】:
-
请注意,在二进制分类中,pbs(不一定不平衡)this 可能会发生在
SVC()和一般非概率分类器中。 -
请展示,不要告诉 - 发布minimal reproducible example。
-
正如在下面的答案中添加的那样,文档中可能会出现不一致结果的警告:scikit-learn.org/stable/modules/generated/sklearn.svm.SVC.html。
标签: python scikit-learn classification