From 697cc629a1b9bf9028e31072db3d985bbf82c9dc Mon Sep 17 00:00:00 2001 From: Anson Kahng Date: Fri, 12 Dec 2014 14:48:38 -0500 Subject: [PATCH] predict v2 --- predict_message.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/predict_message.py b/predict_message.py index e23f36a..a077c74 100644 --- a/predict_message.py +++ b/predict_message.py @@ -22,10 +22,10 @@ def predict_message(message): predictions = {'bow':[],'ngram':[]} for classifier in bow_classifiers: - predictions['bow'].append(classifier.predict(bow_message)) + predictions['bow'].extend(classifier.predict(bow_message)) for classifier in ngram_classifiers: - predictions['ngram'].append(classifier.predict(ngram_message)) + predictions['ngram'].extend(classifier.predict(ngram_message)) # print classifier.predict(ngram_message) print json.dumps(predictions)