def test_evaluate_accuracy_nn(): """Test evaluate_accuracy for nn model""" classifer = ReviewClassifier('nn') classifer.load_model(tar_file='test_nn_model.tar') score = classifer.evaluate_accuracy('test-reviews.csv') assert score > 0 os.remove('output.log')
def test_evaluate_accuracy_rf(): """Test evaluate_accuracy for rf model""" classifer = ReviewClassifier('rf') classifer.load_model(pickle_file='test_rf_model.pickle') score = classifer.evaluate_accuracy('test-reviews.csv') assert score > 0 os.remove('output.log')