def __init__(self): self.cleaner = Cleaner() self.feature_tfidf = FeatureVectorTfIdf() self.classifier = NaiveBayes()
#!/usr/bin/env python from __future__ import division from analysis.preprocess import Cleaner from analysis.learning import FeatureVectorTfIdf, NaiveBayes __author__ = 'Md Mehrab Alam' cleaner = Cleaner() feature_tfidf = FeatureVectorTfIdf() classifier = NaiveBayes() class TrainingAndTesting(object): def __init__(self): self.cleaner = Cleaner() self.feature_tfidf = FeatureVectorTfIdf() self.classifier = NaiveBayes() def train_me(self, traning_data, traning_data_type): data_clean = self.cleaner.lower_and_replace(traning_data) t = self.feature_tfidf.extract_features(data_clean) self.classifier.training(t, traning_data_type) def test_me(self, test_data): data_clean = self.cleaner.lower_and_replace(test_data) t = self.feature_tfidf.word_vectors_tfidf(data_clean) result = self.classifier.testing(t) return result @classmethod