python nltk天真的贝叶斯概率

时间:2013-12-25 13:17:01

标签: python text classification nltk

有没有办法使用nltk.NaiveBayesClassifier.classify获取个体概率?我想看看分类的可能性,试图建立一个置信度量表。显然,使用二元分类器,决策将是一个或另一个,但有没有办法看到决策是如何做出的内部工作?或者,我只需要编写自己的分类器吗?

由于

1 个答案:

答案 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)))