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
