def run_classifier(self, method, classifier):
if method == 'classifier':
self.classifier = Classifier(
self.prepro.severity.keys(), self.prepro.X_train,
self.prepro.y_train, self.prepro.X_test, self.prepro.y_test,
self.prepro.train_size, self.prepro.test_size)
self.classifier.classify(classifier)
if method == 'pipeline':
self.classifier = PipeLineClassifier(
self.prepro.severity.keys(), self.prepro.train_corpus,
self.prepro.y_train, self.prepro.X_test, self.prepro.y_test,
self.prepro.train_size, self.prepro.test_size)
self.classifier.setVariables(classifier)
self.classifier.benchmark()
