Spaces:
Sleeping
Sleeping
hichem-abdellali
commited on
Commit
•
fa350f6
1
Parent(s):
22a0855
set two graphs recognition ratios and recognition summary
Browse files- user-friendly-metrics.py +9 -4
user-friendly-metrics.py
CHANGED
@@ -135,17 +135,22 @@ class UserFriendlyMetrics(evaluate.Metric):
|
|
135 |
)
|
136 |
|
137 |
categories = {
|
138 |
-
"
|
139 |
-
|
140 |
-
|
|
|
141 |
"recognition_0.3",
|
142 |
"recognition_0.5",
|
143 |
"recognition_0.8",
|
144 |
},
|
145 |
-
"
|
|
|
|
|
|
|
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 |
|