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