Expose use_effective_order flag for bleu score computation for shorter sentences

PiperOrigin-RevId: 642433884

Author: T5 Team

t5/evaluation/metrics.py

@@ -45,14 +45,15 @@
 ModelOutputType = seqio.metrics.ModelOutputType
 
 
-def bleu(targets, predictions, tokenizer="intl"):
+def bleu(targets, predictions, tokenizer="intl", use_effective_order=False):
   """Computes BLEU score.
 
   Args:
     targets: list of strings or list of list of strings if multiple references
       are present.
     predictions: list of strings
     tokenizer: tokenizer option for corpus_bleu
+    use_effective_order: whether to use effective order when computing BLEU.
 
   Returns:
     bleu_score across all targets and predictions
@@ -69,7 +70,7 @@ def bleu(targets, predictions, tokenizer="intl"):
                                      force=False,
                                      lowercase=False,
                                      tokenize=tokenizer,
-                                     use_effective_order=False)
+                                     use_effective_order=use_effective_order)
   return {"bleu": bleu_score.score}