float, int results

user-friendly-metrics.py

@@ -149,14 +149,14 @@ def calculate(predictions,
     unique_gt_ids = num_gt_ids(df)
 
     for key in summary:
-        summary[key] = summary[key][0]
+        summary[key] = float(summary[key][0])
 
     summary["num_gt_ids"] = unique_gt_ids
 
     for th in recognition_thresholds:
         recognized = recognition(tr_ratios, th)
-        summary[f'recognition_{th}'] = recognition(tr_ratios, th)/unique_gt_ids
-        summary[f'recognized_{th}'] = recognized
+        summary[f'recognition_{th}'] = float(recognition(tr_ratios, th)/unique_gt_ids)
+        summary[f'recognized_{th}'] = int(recognized)
     
     return summary