未验证 提交 ec1c88ae 编写于 作者: H Hui Zhang 提交者: GitHub

[s2t] remove nltk (#1332)

上级 6ba1f16d
...@@ -14,7 +14,6 @@ ...@@ -14,7 +14,6 @@
"""This module provides functions to calculate bleu score in different level. """This module provides functions to calculate bleu score in different level.
e.g. wer for word-level, cer for char-level. e.g. wer for word-level, cer for char-level.
""" """
import nltk
import numpy as np import numpy as np
import sacrebleu import sacrebleu
...@@ -114,6 +113,5 @@ class ErrorCalculator(): ...@@ -114,6 +113,5 @@ class ErrorCalculator():
seq_true_text = "".join(seq_true).replace(self.space, " ") seq_true_text = "".join(seq_true).replace(self.space, " ")
seqs_hat.append(seq_hat_text) seqs_hat.append(seq_hat_text)
seqs_true.append(seq_true_text) seqs_true.append(seq_true_text)
bleu = nltk.bleu_score.corpus_bleu([[ref] for ref in seqs_true], bleu = sacrebleu.corpus_bleu(seqs_hat, [[ref] for ref in seqs_true])
seqs_hat) return bleu.score * 100
return bleu * 100
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册