Gil-Simas commited on
Commit
fb21e6c
·
1 Parent(s): 016b10d

zero div handled

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +4 -3
user-friendly-metrics.py CHANGED
@@ -307,9 +307,10 @@ 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
- print("Precision: ", metrics_dict["precision"])
311
- print("Recall: ", metrics_dict["recall"])
312
- metrics_dict["f1"] = 2*metrics_dict["precision"]*metrics_dict["recall"]/(metrics_dict["precision"]+metrics_dict["recall"])
 
313
 
314
  for th in recognition_thresholds:
315
  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
+ 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"]