ibrahimsharaf#14 added classifier method test_model

Ayatallah · May 14, 2019 · d9dcbfa · d9dcbfa
1 parent 67d6ee9
commit d9dcbfa
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/models/classifier_builder.py b/models/classifier_builder.py
@@ -33,4 +33,12 @@ def load_model(self,filename):
             loaded_model = joblib.load(filename)
             self.model = loaded_model
         else:
-            self.model = None
+            self.model = None
+
+    def test_model(self,d2v, testing_vectors, testing_labels):
+        logging.info("Classifier testing")
+        test_vectors = doc2VecBuilder.get_vectors(d2v, len(testing_vectors), 300, 'Test')
+        testing_predictions = self.model.predict(test_vectors)
+        logging.info('Testing predicted classes: {}'.format(np.unique(testing_predictions)))
+        logging.info('Testing accuracy: {}'.format(accuracy_score(testing_labels, testing_predictions)))
+        logging.info('Testing F1 score: {}'.format(f1_score(testing_labels, testing_predictions, average='weighted')))