torcheval.metrics.functional.word_error_rate

torcheval.metrics.functional.word_error_rate(input: str | List[str], target: str | List[str]) → Tensor

计算预测单词序列与参考单词序列的词错误率。 其类版本为torcheval.metrics.WordErrorRate。

Parameters:

• input (str, List[str]) – 要评分的预测单词序列，可以是字符串或字符串列表。

• target (str, List[str]) – 参考词序列，可以是字符串或字符串列表。

示例

>>> import torch
>>> from torcheval.metrics.functional import word_error_rate
>>> input = ["hello world", "welcome to the facebook"]
>>> target = ["hello metaverse", "welcome to meta"]
>>> word_error_rate(input, target)
tensor(0.6)
>>> input = ["this is the prediction", "there is an other sample"]
>>> target = ["this is the reference", "there is another one"]
>>> word_error_rate(input, target)
tensor(0.5)