[s2t] remove nltk (#1332)

pull/1334/head
Hui Zhang 4 years ago committed by GitHub
parent 6ba1f16d13
commit ec1c88ae1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save