首页 文章

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

提问于
浏览
1

我正在制作一个机器学习程序,它对以下类别之一的单词进行分类:硬件,软件,None_of_these . 我使用sklearn的Multinomial Naive Bayes分类器 .

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

是否有一个函数可以给出每个样本的概率?

1 回答

  • 0

    没关系,我找到了解决方案:

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

相关问题