def train(self, labels, train_set): data = helper.format_for_nltk(labels, train_set) self.classifier = nb.NaiveBayesClassifier.train(data)
def train(self, labels, train_set): data = helper.format_for_nltk(labels, train_set) self.classifier = maxent.MaxentClassifier.train(data) # TODO use this in the main.py self.classifier.show_most_informative_features(5)