sklearn NB分类器：如何获得单个样本的实际概率？

Question

我正在制作一个机器学习程序，它对以下类别之一的单词进行分类：硬件，软件，无法使用。我使用sklearn的Multinomial Naive Bayes分类器。

函数predict（）给出了每个单词的预测，但是，我无法看到该单词与预测的类别匹配的实际概率（浮点范围为0到1.0）。我在sklearn的网站上也没有找到这个。

是否有一个函数可以提供每个样本的概率？

Answer 1

没关系，我找到了解决办法。：

predict_proba（X）返回测试向量X的概率估计值。