Spaces:
Sleeping
Sleeping
add eps to handle 0 div
Browse files- 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 |
-
|
311 |
-
|
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"]
|