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!
2 files
+ 15
15
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 1
1
@@ -94,7 +94,7 @@ def print_worst_predictions(all_inferences: Dict[str, List[Inference]]):
inference.prediction,
)
alignment_str = f'{alignment["query_aligned"]}\n{alignment["matched_aligned"]}\n{alignment["target_aligned"]}'
table.add_row([inference.image, inference.wer, alignment_str])
table.add_row([inference.image, round(inference.wer * 100, 2), alignment_str])
print(f"\n#### {NB_WORST_PREDICTIONS} worst prediction(s)\n")
print(table)
Loading