Gil-Simas commited on
Commit
0c6478d
·
1 Parent(s): 8455f0e

add unique ids

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +2 -0
user-friendly-metrics.py CHANGED
@@ -150,6 +150,8 @@ def calculate(predictions,
150
  for key in summary:
151
  summary[key] = summary[key][0]
152
 
 
 
153
  for th in recognition_thresholds:
154
  summary[f'recognition_{th}'] = recognition(tr_ratios, th)/unique_gt_ids
155
 
 
150
  for key in summary:
151
  summary[key] = summary[key][0]
152
 
153
+ summary["num_gt_ids"] = unique_gt_ids
154
+
155
  for th in recognition_thresholds:
156
  summary[f'recognition_{th}'] = recognition(tr_ratios, th)/unique_gt_ids
157