Add train accuracy to accuracy test

Jamesm4 · May 4, 2018 · 082cbb8 · 082cbb8
1 parent 5007980
commit 082cbb8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/__pycache__/sentiment.cpython-35.pyc b/__pycache__/sentiment.cpython-35.pyc
diff --git a/classifier.py b/classifier.py
@@ -26,10 +26,14 @@ def n_classify(self, text):
     return json.dumps(probs)
 
   def accuracy(self):
+
+    fp = open('./data/train.csv')
+    train_accuracy = self.cl.accuracy(fp, format="csv")
+    fp.close()
     fp = open('./data/test.csv')
     test_accuracy = self.cl.accuracy(fp, format="csv")
     fp.close()
-    return test_accuracy
+    return json.dumps({"train_accuracy": train_accuracy, "test_accuracy": test_accuracy})
 
   def labels(self):
     return json.dumps({"labels": self.cl.labels()})

diff --git a/sentiment.py b/sentiment.py
@@ -24,7 +24,7 @@ def sentiments():
 def accuracy():
   acc = cache.get("accuracy")
   if acc is None:
-    acc = str(round(cl.accuracy(), 3))
+    acc = cl.accuracy()
     cache.set("accuracy", acc)
   return acc