Spaces:
Sleeping
Sleeping
only pr rec
Browse files- 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))
|