Spaces:
Sleeping
Sleeping
bug
Browse files- user-friendly-metrics.py +4 -1
user-friendly-metrics.py
CHANGED
@@ -154,7 +154,10 @@ def calculate(predictions,
|
|
154 |
unique_gt_ids = num_gt_ids(df)
|
155 |
|
156 |
for key in summary:
|
157 |
-
|
|
|
|
|
|
|
158 |
|
159 |
summary["num_gt_ids"] = unique_gt_ids
|
160 |
|
|
|
154 |
unique_gt_ids = num_gt_ids(df)
|
155 |
|
156 |
for key in summary:
|
157 |
+
if key in namemap:
|
158 |
+
summary[namemap[key]] = float(summary[key][0])
|
159 |
+
else:
|
160 |
+
summary[key] = float(summary[key][0])
|
161 |
|
162 |
summary["num_gt_ids"] = unique_gt_ids
|
163 |
|