我正在制作一个机器学习程序,它对以下类别之一的单词进行分类:硬件,软件,None_of_these . 我使用sklearn的Multinomial Naive Bayes分类器 .
函数predict()给出了每个单词的预测,但是,我无法看到该单词与预测的类别匹配的实际概率(浮点范围为0到1.0) . 我也没有在sklearn的网站上找到这个 .
是否有一个函数可以给出每个样本的概率?
没关系,我找到了解决方案:
predict_proba(X)返回测试向量X的概率估计值 .
1 回答
没关系,我找到了解决方案:
predict_proba(X)返回测试向量X的概率估计值 .