From ec1c88ae1a4dc029802b1a9518907902c0e261a0 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Wed, 12 Jan 2022 19:56:58 +0800 Subject: [PATCH] [s2t] remove nltk (#1332) --- paddlespeech/s2t/utils/bleu_score.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/paddlespeech/s2t/utils/bleu_score.py b/paddlespeech/s2t/utils/bleu_score.py index ea32fcf9..a50c000a 100644 --- a/paddlespeech/s2t/utils/bleu_score.py +++ b/paddlespeech/s2t/utils/bleu_score.py @@ -14,7 +14,6 @@ """This module provides functions to calculate bleu score in different level. e.g. wer for word-level, cer for char-level. """ -import nltk import numpy as np import sacrebleu @@ -114,6 +113,5 @@ class ErrorCalculator(): seq_true_text = "".join(seq_true).replace(self.space, " ") seqs_hat.append(seq_hat_text) seqs_true.append(seq_true_text) - bleu = nltk.bleu_score.corpus_bleu([[ref] for ref in seqs_true], - seqs_hat) - return bleu * 100 + bleu = sacrebleu.corpus_bleu(seqs_hat, [[ref] for ref in seqs_true]) + return bleu.score * 100 -- GitLab