Skip to content
Snippets Groups Projects

Display 5 worst predictions at the end of evaluation

Merged Manon Blanco requested to merge display-worst-predictions into main
All threads resolved!
4 files
+ 69
3
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -26,8 +26,10 @@ METRICS_KEYWORD = {"cer": "chars", "wer": "words", "ner": "tokens"}
@dataclass
class Inference:
image: str
ground_truth: str
prediction: str
wer: float
class MetricManager:
Loading