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