From b84e7e150f59b41a18a4eee99468cd89c857b5fc Mon Sep 17 00:00:00 2001 From: Nik Vaessen Date: Sat, 1 Feb 2025 16:58:08 -0500 Subject: [PATCH] replace depricated methods of jiwer --- metrics/cer/cer.py | 14 +++++++------- metrics/wer/wer.py | 10 +++++----- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/metrics/cer/cer.py b/metrics/cer/cer.py index c5f4a9072..1f6269a5c 100644 --- a/metrics/cer/cer.py +++ b/metrics/cer/cer.py @@ -137,23 +137,23 @@ def _info(self): def _compute(self, predictions, references, concatenate_texts=False): if concatenate_texts: - return jiwer.compute_measures( + return jiwer.process_words( references, predictions, - truth_transform=cer_transform, + reference_transform=cer_transform, hypothesis_transform=cer_transform, - )["wer"] + ).wer incorrect = 0 total = 0 for prediction, reference in zip(predictions, references): - measures = jiwer.compute_measures( + out = jiwer.process_words( reference, prediction, - truth_transform=cer_transform, + reference_transform=cer_transform, hypothesis_transform=cer_transform, ) - incorrect += measures["substitutions"] + measures["deletions"] + measures["insertions"] - total += measures["substitutions"] + measures["deletions"] + measures["hits"] + incorrect += out.substitutions + out.deletions + out.insertions + total += out.substitutions + out.deletions + out.hits return incorrect / total diff --git a/metrics/wer/wer.py b/metrics/wer/wer.py index 214d5b22e..87a95d0f7 100644 --- a/metrics/wer/wer.py +++ b/metrics/wer/wer.py @@ -14,7 +14,7 @@ """ Word Error Ratio (WER) metric. """ import datasets -from jiwer import compute_measures +from jiwer import process_words import evaluate @@ -95,12 +95,12 @@ def _info(self): def _compute(self, predictions=None, references=None, concatenate_texts=False): if concatenate_texts: - return compute_measures(references, predictions)["wer"] + return process_words(references, predictions).wer else: incorrect = 0 total = 0 for prediction, reference in zip(predictions, references): - measures = compute_measures(reference, prediction) - incorrect += measures["substitutions"] + measures["deletions"] + measures["insertions"] - total += measures["substitutions"] + measures["deletions"] + measures["hits"] + out = process_words(reference, prediction) + incorrect += out.substitutions + out.deletions + out.insertions + total += out.substitutions + out.deletions + out.hits return incorrect / total