class Hammie(hammie.Hammie):
def __init__(self, bayes):
hammie.Hammie.__init__(self, bayes)
self.tokenizer = Tokenizer()
def _scoremsg(self, msg, evidence=False):
return self.bayes.spamprob(self.tokenizer.tokenize(msg), evidence)
def train(self, msg, is_spam, add_header=False):
self.bayes.learn(self.tokenizer.tokenize(msg), is_spam)
