F1 calculation

urialon · web-flow · commit e7547de9febe · 2022-02-02T14:33:34.000-05:00
return 0 if precision and recall are zero
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:
