fix variable naming error
Browse files- 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(
|
108 |
-
bleu2 = bleu(
|
109 |
-
bleu3 = bleu(
|
110 |
-
bleu4 = bleu(
|
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,
|