From ca8fa90b0207809b43052601c930ad1666542d3a Mon Sep 17 00:00:00 2001 From: smallv0221 <33639025+smallv0221@users.noreply.github.com> Date: Wed, 16 Dec 2020 13:08:40 +0800 Subject: [PATCH] fix bleu bug (#5065) --- 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 21cce4b9..0cf261ce 100644 --- a/PaddleNLP/paddlenlp/metrics/bleu.py +++ b/PaddleNLP/paddlenlp/metrics/bleu.py @@ -248,7 +248,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 @@ -258,6 +258,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