File size: 507 Bytes
c668e80 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from .scorer import Scorer
from onmt.scorers import register_scorer
from sacrebleu import corpus_bleu
@register_scorer(metric="BLEU")
class BleuScorer(Scorer):
"""BLEU scorer class."""
def __init__(self, opts):
"""Initialize necessary options for sentencepiece."""
super().__init__(opts)
def compute_score(self, preds, texts_refs):
if len(preds) > 0:
score = corpus_bleu(preds, [texts_refs]).score
else:
score = 0
return score
|