model stats

epdb/models.py

@@ -2815,6 +2815,21 @@ class PackageBasedModel(EPModel):
     )
     multigen_eval = models.BooleanField(null=False, blank=False, default=False)
 
+    def statistics(self):
+        return {
+            'accuracy': [
+                self.eval_results["average_accuracy"],
+                self.eval_results.get("multigen_average_accuracy")],
+            'precision': [
+                self.eval_results["average_precision_per_threshold"][f"{self.threshold:.2f}"],
+                self.eval_results.get("multigen_average_precision_per_threshold", {}).get(f"{self.threshold:.2f}")
+            ],
+            'recall': [
+                self.eval_results["average_recall_per_threshold"][f"{self.threshold:.2f}"],
+                self.eval_results.get("multigen_average_recall_per_threshold", {}).get(f"{self.threshold:.2f}")
+            ],
+        }
+
     @property
     def pr_curve(self):
         if self.model_status != self.FINISHED: