Avoid div by 0 errors

Author: mariya

src/conformist/roc.py

@@ -49,8 +49,9 @@ def uncalibrated_model_rates(self, alphas):
             # Get sums of tps and fns
             tps = tps.sum()
             fns = fns.sum()
+            tpr = tps / (tps + fns)
 
-            model_tpr.append(tps / (tps + fns))
+            model_tpr.append(tpr)
 
             # FPS calculation
             fps = ((prediction_sets * (1 - self.prediction_dataset.labels_idx)).sum(axis=1))
@@ -62,8 +63,9 @@ def uncalibrated_model_rates(self, alphas):
             # Get sums of tns and fps
             tns = tns.sum()
             fps = fps.sum()
+            fpr = fps / (fps + tns)
 
-            model_fpr.append(fps / (fps + tns))
+            model_fpr.append(fpr)
 
         return model_tpr, model_fpr
 

src/conformist/validation_run.py

@@ -87,6 +87,9 @@ def FPR(self):
         tns = tns.sum()
         fps = fps.sum()
 
+        if (fps + tns) == 0:
+            tns = self.EPSILON
+
         return fps / (fps + tns)