Gil-Simas commited on
Commit
3e685a7
·
1 Parent(s): 69e2803

another one

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +1 -1
user-friendly-metrics.py CHANGED
@@ -142,7 +142,7 @@ def calculate(predictions,
142
  acc.update(refs[:,0].astype('int').tolist(), preds[:,0].astype('int').tolist(), C)
143
 
144
  mh = mm.metrics.create()
145
- summary = mh.compute(acc, metrics=['recall', 'precision', "num_objects", "num_false_positives", "num_matches", "num_misses", "num_detections"]).to_dict()
146
 
147
  df = events_to_df_map(acc.events)
148
  tr_ratios = track_ratios(df, obj_frequencies(df))
 
142
  acc.update(refs[:,0].astype('int').tolist(), preds[:,0].astype('int').tolist(), C)
143
 
144
  mh = mm.metrics.create()
145
+ summary = mh.compute(acc, metrics=['recall', 'precision', "num_objects", "num_false_positives", "num_matches", "num_misses", "num_detections", "num_predictions"]).to_dict()
146
 
147
  df = events_to_df_map(acc.events)
148
  tr_ratios = track_ratios(df, obj_frequencies(df))