hichem-abdellali commited on
Commit
fa350f6
1 Parent(s): 22a0855

set two graphs recognition ratios and recognition summary

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +9 -4
user-friendly-metrics.py CHANGED
@@ -135,17 +135,22 @@ class UserFriendlyMetrics(evaluate.Metric):
135
  )
136
 
137
  categories = {
138
- "confusion_metrics": {"fp", "tp", "fn"},
139
- "evaluation_metrics": {"f1", "recall", "precision"},
140
- "recognition_metrics": {
 
141
  "recognition_0.3",
142
  "recognition_0.5",
143
  "recognition_0.8",
144
  },
145
- "recognized_metrics": {
 
 
 
146
  "recognized_0.3",
147
  "recognized_0.5",
148
  "recognized_0.8",
 
149
  },
150
  }
151
 
 
135
  )
136
 
137
  categories = {
138
+ "recognition_ratios": {
139
+ "f1",
140
+ "recall",
141
+ "precision",
142
  "recognition_0.3",
143
  "recognition_0.5",
144
  "recognition_0.8",
145
  },
146
+ "recognition_summary": {
147
+ "fp",
148
+ "tp",
149
+ "fn",
150
  "recognized_0.3",
151
  "recognized_0.5",
152
  "recognized_0.8",
153
+ "num_gt_ids",
154
  },
155
  }
156