def main():
train_email_data = EmailData()
train_email_data.load_from_file('data/train')
feature = Feature()
feature.learn(train_email_data)
train_data_set = feature.translate_email_data(train_email_data)
#print(feature.features)
naive_bayesian = NaiveBayesian()
naive_bayesian.learn(feature, train_data_set)
test_email_data = EmailData()
test_email_data.load_from_file('data/test')
test_data_set = feature.translate_email_data(test_email_data)
print('# Training set')
test(naive_bayesian, train_data_set)
print('# Testing set')
test(naive_bayesian, test_data_set)
