有没有办法使用nltk.NaiveBayesClassifier.classify获取个体概率?我想看看分类的可能性,试图建立一个置信度量表。显然,使用二元分类器,决策将是一个或另一个,但有没有办法看到决策是如何做出的内部工作?或者,我只需要编写自己的分类器吗?
由于
答案 0 :(得分:14)
nltk.NaiveBayesClassifier.prob_classify
怎么样?
http://nltk.org/api/nltk.classify.html#nltk.classify.naivebayes.NaiveBayesClassifier.prob_classify
classify
调用此函数:
def classify(self, featureset):
return self.prob_classify(featureset).max()
编辑:这样的事情应该有效(未经测试):
dist = classifier.prob_classify(features)
for label in dist.samples():
print("%s: %f" % (label, dist.prob(label)))