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)