Ian Borrego Obrador commited on
Commit
1b5f36c
·
1 Parent(s): a63673d

small tokenizer mod for bleu

Browse files
Files changed (1) hide show
  1. generation_evaluator.py +8 -3
generation_evaluator.py CHANGED
@@ -174,9 +174,14 @@ class GenerationEvaluator(evaluate.Metric):
174
  )
175
 
176
  # Compute BLEU
177
- bleu_results = self.bleu_scorer.compute(
178
- predictions=predictions, references=references, tokenizer=tokenizer
179
- )
 
 
 
 
 
180
 
181
  # Compute Exact Match
182
  exact_match_results = self.exact_match_scorer.compute(
 
174
  )
175
 
176
  # Compute BLEU
177
+ if tokenizer is None:
178
+ bleu_results = self.bleu_scorer.compute(
179
+ predictions=predictions, references=references
180
+ )
181
+ else:
182
+ bleu_results = self.bleu_scorer.compute(
183
+ predictions=predictions, references=references, tokenizer=tokenizer
184
+ )
185
 
186
  # Compute Exact Match
187
  exact_match_results = self.exact_match_scorer.compute(