Spaces:
Sleeping
Sleeping
zero div handled
Browse files- 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 |
-
|
311 |
-
|
312 |
-
|
|
|
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"]
|