Gil-Simas commited on
Commit
35a2b6a
·
1 Parent(s): 0c6478d

only pr rec

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