You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
768 B
Python
24 lines
768 B
Python
5 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from nltk.classify.naivebayes import NaiveBayesClassifier
|
||
|
|
|
||
|
|
|
||
|
|
class NaiveBayesClassifierTest(unittest.TestCase):
|
||
|
|
def test_simple(self):
|
||
|
|
training_features = [
|
||
|
|
({'nice': True, 'good': True}, 'positive'),
|
||
|
|
({'bad': True, 'mean': True}, 'negative'),
|
||
|
|
]
|
||
|
|
|
||
|
|
classifier = NaiveBayesClassifier.train(training_features)
|
||
|
|
|
||
|
|
result = classifier.prob_classify({'nice': True})
|
||
|
|
self.assertTrue(result.prob('positive') > result.prob('negative'))
|
||
|
|
self.assertEqual(result.max(), 'positive')
|
||
|
|
|
||
|
|
result = classifier.prob_classify({'bad': True})
|
||
|
|
self.assertTrue(result.prob('positive') < result.prob('negative'))
|
||
|
|
self.assertEqual(result.max(), 'negative')
|