From 4dca825aec213e6faef72716ef6b975961eb524c Mon Sep 17 00:00:00 2001 From: smallv0221 <397551318@qq.com> Date: Wed, 16 Dec 2020 03:35:53 +0000 Subject: [PATCH] fix bleu bug --- PaddleNLP/paddlenlp/metrics/bleu.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PaddleNLP/paddlenlp/metrics/bleu.py b/PaddleNLP/paddlenlp/metrics/bleu.py index af5f6811..955ea86f 100644 --- a/PaddleNLP/paddlenlp/metrics/bleu.py +++ b/PaddleNLP/paddlenlp/metrics/bleu.py @@ -174,7 +174,7 @@ class BLEUForDuReader(BLEU): ref_ngram = [] for ref_id, r in enumerate(yn_ref): ref_ngram.append(get_ngram(ref_list[ref_id], n_size, label=r)) - match_size, cand_size = self.get_match_size(cand_ngram, ref_ngram) + match_size, cand_size = get_match_size(cand_ngram, ref_ngram) self.match_ngram[n_size] += self.alpha * match_size self.candi_ngram[n_size] += self.alpha * match_size @@ -184,6 +184,6 @@ class BLEUForDuReader(BLEU): ref_ngram = [] for reff_id, r in enumerate(entity_ref): ref_ngram.append(get_ngram(r, n_size, label='ENTITY')) - match_size, cand_size = self.get_match_size(cand_ngram, ref_ngram) + match_size, cand_size = get_match_size(cand_ngram, ref_ngram) self.match_ngram[n_size] += self.beta * match_size self.candi_ngram[n_size] += self.beta * match_size -- GitLab