Sklearn 朴素贝叶斯打印概率结果

时间:2021-04-22 06:20:50

标签: python scikit-learn naivebayes

分类器的输出基本上是 1 和 0 取决于转换后的标签,对吗? 但是如何打印出它的计算结果呢? 而不是,

预测:1

1 表示高,0 表示低

我想要输出

预测:86.4% 低

如何打印其百分比?

1 个答案:

答案 0 :(得分:0)

尝试使用模型的方法:predict_proba(X)

此函数将返回模型中每个类的样本概率。例如,如果您有一个二元分类问题,您的输出将类似于:

[0.22 , 0.78]

这意味着您测试的样本有 22% 可能属于 0 类,78% 可能属于 1 类。因此,您正在获得属于每个类的百分比。