F1 calculation

return 0 if precision and recall are zero
tech-srl · Feb 2, 2022 · e7547de · e7547de
1 parent f33552e
commit e7547de
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tensorflow_model.py b/tensorflow_model.py
@@ -490,7 +490,10 @@ def recall(self):
 
     @property
     def f1(self):
-        return 2 * self.precision * self.recall / (self.precision + self.recall)
+        if self.precision + self.recall == 0:
+            return 0
+        else:
+            return 2 * self.precision * self.recall / (self.precision + self.recall)
 
 
 class TopKAccuracyEvaluationMetric: