(#2) Victim: Add evaluation text

betarixm · Apr 18, 2022 · a9b9250 · a9b9250
1 parent ce6378a
commit a9b9250
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/victim/train.py b/src/victim/train.py
@@ -60,6 +60,10 @@
         )
 
     if args.evaluate:
-        print(classifier.evaluate())
+        result = classifier.evaluate()
+        print(f"[*] Evaluation of {args.dataset.upper()} Classifier")
+        for key in result:
+            print(f"    - {key}: {result[key]}")
+
     else:
         classifier.train(e)