Gil-Simas commited on
Commit
c356001
·
1 Parent(s): 432dae9

add num objects

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']).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"]).to_dict()
146
 
147
  df = events_to_df_map(acc.events)
148
  tr_ratios = track_ratios(df, obj_frequencies(df))