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

Question

我正在制作一个机器学习程序，它将单词分类为以下类别之一：硬件、软件、None_of_these。我使用了 sklearn 的多项朴素贝叶斯分类器。

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

有没有一个函数可以给我每个样本的概率？

Answer 1

没关系，我找到了解决方案。：

predict_proba(X) 返回测试向量 X 的概率估计值。