Add more metrics

audioshake · Jul 9, 2024 · 2aa774b · 2aa774b
1 parent b8f77b5
commit 2aa774b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/alt_eval/metrics.py b/src/alt_eval/metrics.py
@@ -125,7 +125,12 @@ def compute_word_metrics(
         "WER": wo.wer,
         "MER": wo.mer,
         "WIL": wo.wil,
+        "hits": wo.hits,
+        "substitutions": wo.substitutions,
+        "deletions": wo.deletions,
+        "insertions": wo.insertions,
         "ER_case": error_counts["case"] / total_len,
+        "WER_case": wo.wer + error_counts["case"] / total_len,
     }
     if visualize_errors:
         results["errors_html"] = vis_htmls