add missing condition for lesion ppv and f1_score

ivadomed · Oct 18, 2024 · 250983f · 250983f
1 parent e093be2
commit 250983f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/MetricsReloaded/metrics/pairwise_measures.py b/MetricsReloaded/metrics/pairwise_measures.py
@@ -1257,6 +1257,9 @@ def lesion_f1_score(self):
         elif not self.flag_empty_ref and self.flag_empty_pred:
             # Reference is not empty, prediction is empty --> model did not learn correctly (it's false negative)
             return 0.0
+        elif self.flag_empty_ref and not self.flag_empty_pred:
+            # Reference is empty, prediction is not empty --> model did not learn correctly (it's false positive)
+            return 0.0
         # if the predction is not empty and ref is empty, it's false positive
         # if both are not empty, it's true positive
         else:
@@ -1285,6 +1288,9 @@ def lesion_ppv(self):
         elif not self.flag_empty_ref and self.flag_empty_pred:
             # Reference is not empty, prediction is empty --> model did not learn correctly (it's false negative)
             return 0.0
+        elif self.flag_empty_ref and not self.flag_empty_pred:
+            # Reference is empty, prediction is not empty --> model did not learn correctly (it's false positive)
+            return 0.0
         # if the predction is not empty and ref is empty, it's false positive
         # if both are not empty, it's true positive
         else: