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
+ 33
26
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 8
1
@@ -89,7 +89,14 @@ def eval_nerval(
if not (ground_truths and predictions):
continue
scores = evaluate(ground_truths, predictions, threshold)
scores = {
key: {
k: round(value * 100, 2) if k in ["P", "R", "F1"] else value
for k, value in values.items()
}
for key, values in evaluate(ground_truths, predictions, threshold).items()
}
print(f"\n##### {split_name}\n")
print_results(scores)
Loading