Skip to content
Snippets Groups Projects

Fix case with empty GT

Merged Solene Tarride requested to merge fix-metric-computation into main
All threads resolved!
1 file
+ 2
0
Compare changes
  • Side-by-side
  • Inline
@@ -206,6 +206,8 @@ class MetricManager:
metrics["nb_" + keyword + suffix] = list(map(len, split_gt))
metrics[keyword + "_error_rate" + suffix] = [
round(float(edit_dist / gt_len), 4)
if gt_len != 0
else float(edit_dist == 0)
for edit_dist, gt_len in zip(
metrics["edit_" + keyword + suffix],
metrics["nb_" + keyword + suffix],
)
]
case "loss" | "loss_ce":
metrics[metric_name] = [
values[metric_name],
Loading