Store the accuracy data per project as JSON when manually evaluating.

WuXiangChen · Apr 26, 2021 · 50ebbb3 · 50ebbb3
1 parent c515304
commit 50ebbb3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Util/sample_links.py b/Util/sample_links.py
@@ -1,3 +1,4 @@
+import json
 import random
 import sys
 import webbrowser as wb
@@ -62,3 +63,6 @@ def prompt_user_bool(prompt_msg, default=None, options_map=None):
 
     for project in projects:
         print('The GT accuracy over the sample is %2.3f' % np.mean(hits[project]))
+
+    with open('./gt_accuracy.json', 'w') as f:
+        f.write(json.dumps(hits))
diff --git a/Util/sample_predictions.py b/Util/sample_predictions.py
@@ -1,4 +1,5 @@
 import ast
+import json
 import random
 import webbrowser as wb
 
@@ -63,3 +64,6 @@
 
     for project in projects:
         print('The GT accuracy over the sample is %2.3f' % np.mean(hits[project]))
+
+    with open('./gt_accuracy.json', 'w') as f:
+        f.write(json.dumps(hits))