Gil-Simas commited on
Commit
5591ba8
·
1 Parent(s): fb21e6c

add eps to handle 0 div

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +2 -4
user-friendly-metrics.py CHANGED
@@ -307,10 +307,8 @@ def realize_metrics(metrics_dict,
307
 
308
  metrics_dict["precision"] = metrics_dict["tp"]/(metrics_dict["tp"]+metrics_dict["fp"])
309
  metrics_dict["recall"] = metrics_dict["tp"]/(metrics_dict["tp"]+metrics_dict["fn"])
310
- if metrics_dict["precision"]+metrics_dict["recall"] == 0:
311
- metrics_dict["f1"] = 0
312
- else:
313
- metrics_dict["f1"] = 2*metrics_dict["precision"]*metrics_dict["recall"]/(metrics_dict["precision"]+metrics_dict["recall"])
314
 
315
  for th in recognition_thresholds:
316
  metrics_dict[f"recognition_{th}"] = metrics_dict[f"recognized_{th}"]/metrics_dict["num_gt_ids"]
 
307
 
308
  metrics_dict["precision"] = metrics_dict["tp"]/(metrics_dict["tp"]+metrics_dict["fp"])
309
  metrics_dict["recall"] = metrics_dict["tp"]/(metrics_dict["tp"]+metrics_dict["fn"])
310
+
311
+ metrics_dict["f1"] = 2*metrics_dict["precision"]*metrics_dict["recall"]/(metrics_dict["precision"]+metrics_dict["recall"]+1e-6)
 
 
312
 
313
  for th in recognition_thresholds:
314
  metrics_dict[f"recognition_{th}"] = metrics_dict[f"recognized_{th}"]/metrics_dict["num_gt_ids"]