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

问题描述

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

预测:1

1 表示高,0 表示低

我想要输出

预测:86.4% 低

如何打印其百分比?

解决方法

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

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

[0.22,0.78]

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