1from naiveBayesClassifier.trainedData import TrainedData 2 3class Trainer(object): 4 5 """docstring for Trainer""" 6 def __init__(self, tokenizer): 7 super(Trainer, self).__init__() 8 self.tokenizer = tokenizer 9 self.data = TrainedData() 10 11 def train(self, text, className): 12 """ 13 enhances trained data using the given text and class 14 """ 15 self.data.increaseClass(className) 16 17 tokens = self.tokenizer.tokenize(text) 18 for token in tokens: 19 self.data.increaseToken(token, className)