首页 > 解决方案 > Sklearn 朴素贝叶斯打印概率结果

问题描述

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

预测:1

1 表示高,0 表示低

我想要输出

预测:86.4% 低

如何打印它的百分比?

标签: pythonscikit-learnnaivebayes

解决方案


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

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

[0.22 , 0.78]

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


推荐阅读