o yi @sddlmZmZmZddlZddlmZmZddlmZdee ee fdee ee fdeeeffdd Z d ed edefd d Z dee ee fdee ee fdefddZ dS))ListTupleUnionN)Tensortensor)_edit_distancepredstargetreturncCst|tr|g}t|tr|g}tdtjd}tdtjd}t||D]\}}|}|}|t||7}|t|7}q#||fS)aUpdate the wer score with the current set of references and predictions. Args: preds: Transcription(s) to score as a string or list of strings target: Reference(s) for each speech input as a string or list of strings Returns: Number of edit operations to get from the reference to the prediction, summed over all samples Number of words overall references r)dtype) isinstancestrrtorchfloatzipsplitrlen)rr errorstotalpredtgt pred_tokens tgt_tokensrT/home/ubuntu/.local/lib/python3.10/site-packages/torchmetrics/functional/text/wer.py _wer_updates  rrrcCs||S)zCompute the word error rate. Args: errors: Number of edit operations to get from the reference to the prediction, summed over all samples total: Number of words overall references Returns: Word error rate score r)rrrrr _wer_compute3s rcCst||\}}t||S)aWord error rate (WordErrorRate_) is a common metric of the performance of an automatic speech recognition system. This value indicates the percentage of words that were incorrectly predicted. The lower the value, the better the performance of the ASR system with a WER of 0 being a perfect score. Args: preds: Transcription(s) to score as a string or list of strings target: Reference(s) for each speech input as a string or list of strings Returns: Word error rate score Examples: >>> preds = ["this is the prediction", "there is an other sample"] >>> target = ["this is the reference", "there is another one"] >>> word_error_rate(preds=preds, target=target) tensor(0.5000) )rr)rr rrrrrword_error_rate@s r) typingrrrrrr#torchmetrics.functional.text.helperrr rrrrrrrs   2