Python PyTorch bleu_score用法及代码示例

本文简要介绍python语言中 torchtext.data.metrics.bleu_score 的用法。

用法:

torchtext.data.metrics.bleu_score(candidate_corpus, references_corpus, max_n=4, weights=[0.25, 0.25, 0.25, 0.25])

参数：

• candidate_corpus-候选翻译的可迭代。每个翻译都是一个可迭代的标记

• references_corpus-参考翻译的可迭代的迭代。每个翻译都是一个可迭代的标记

• max_n-我们要使用的最大值n-gram。例如：如果 max_n=3，我们将使用 unigrams、bigrams 和 trigrams

• weights-用于每个 n-gram 类别的权重列表(默认情况下统一)

计算候选翻译语料库和参考翻译语料库之间的 BLEU 分数。基于https://www.aclweb.org/anthology/P02-1040.pdf

例子

>>> from torchtext.data.metrics import bleu_score
>>> candidate_corpus = [['My', 'full', 'pytorch', 'test'], ['Another', 'Sentence']]
>>> references_corpus = [[['My', 'full', 'pytorch', 'test'], ['Completely', 'Different']], [['No', 'Match']]]
>>> bleu_score(candidate_corpus, references_corpus)
    0.8408964276313782