Ensure no crash in scoring when 0 n_all_positive_instances

eyra · May 9, 2024 · 1cfade1 · 1cfade1
1 parent 1488750
commit 1cfade1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/score.py b/score.py
@@ -83,7 +83,10 @@ def score(prediction_path, ground_truth_path, output):
     except ZeroDivisionError:
         precision = 0
 
-    recall = true_positives / n_all_positive_instances
+    try:
+        recall = true_positives / n_all_positive_instances
+    except ZeroDivisionError:
+        recall = 0
 
     try:
         f1_score = 2 * (precision * recall) / (precision + recall)