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"]
|