print '\n' from Model import Model test_model = Model() test_model.UpdateBaselines(train_reviews) print "Mean: " + str(test_model.mean) print('training model') from nltk.corpus import stopwords test_model.exclude = stopwords.words('english') test_model.Train(train_reviews) print('done, running evaluations.\n') from Evaluation import evaluate_rigorous_dist evaluate_rigorous_dist(test_reviews, test_model.wordReviewNet, modePercentage, test_model) #test_model.Guess(test_reviews[0]["review"], test_reviews[0]["score"], True) ### Generate confusion matrix of scores from sklearn.metrics import confusion_matrix import matplotlib.pyplot as plt score_true = [] score_guess = []