Gil-Simas commited on
Commit
268f02a
·
1 Parent(s): d73e76d
Files changed (1) hide show
  1. 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
- summary[namemap[key]] = float(summary[key][0])
 
 
 
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