fnvls commited on
Commit
a919448
·
1 Parent(s): afd8f7f

fix variable naming error

Browse files
Files changed (1) hide show
  1. bleu_1234.py +4 -4
bleu_1234.py CHANGED
@@ -104,10 +104,10 @@ class bleu_1234(evaluate.Metric):
104
  references = [[ref] for ref in references]
105
 
106
  bleu = BLEUScore()
107
- bleu1 = bleu(preds, target, n_gram=1)
108
- bleu2 = bleu(preds, target, n_gram=2)
109
- bleu3 = bleu(preds, target, n_gram=3)
110
- bleu4 = bleu(preds, target, n_gram=4)
111
 
112
  output_dict = {
113
  "bleu-1": bleu1,
 
104
  references = [[ref] for ref in references]
105
 
106
  bleu = BLEUScore()
107
+ bleu1 = bleu(predictions, references, n_gram=1)
108
+ bleu2 = bleu(predictions, references, n_gram=2)
109
+ bleu3 = bleu(predictions, references, n_gram=3)
110
+ bleu4 = bleu(predictions, references, n_gram=4)
111
 
112
  output_dict = {
113
  "bleu-1": bleu1,