Example #1
0
 def test_spam_probability(self):
     training_set = [("foo bar", 1), ("bar baz", 0)]
     counts = nb.count_words(training_set)
     word_probs = nb.word_probabilities(counts, 1, 1)
     message = "there is no foo bar like bar bar"
     actual = nb.spam_probability(word_probs, message)
     expected = 0.9
     self.assertAlmostEqual(actual, expected)
Example #2
0
 def classify(self, message):
     return nb.spam_probability(self.word_probs, message)