2016-09-27 74 views
0

我正在制作一个机器学习程序,将以下类别之一的字分类:硬件,软件,None_of_these。我利用sklearn中的Multinomial朴素贝叶斯分类器。sklearn NB分类器:如何获得单个样本的实际概率?

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

是否有一个函数给我每个样本的概率?

回答

0

没关系,我找到了解决办法。:

predict_proba(X)返回测试向量X

概率估计