(#2) Victim: Add evaluation

betarixm · Apr 18, 2022 · 8d6100b · 8d6100b
1 parent 60f9f36
commit 8d6100b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/typings/models.py b/src/typings/models.py
@@ -139,10 +139,13 @@ def train(self, epochs: int = 100):
             ],
         )
 
-        self.__model.evaluate(self.data_test)
-
         self.post_train()
 
+    def evaluate(self):
+        self.load()
+        result = self.__model.evaluate(self.data_test)
+        return dict(zip(self.__model.metrics_names, result))
+
 
 class Attack(ABC):
     def __init__(