fix: per_item_dialog_bleu metric usage (#1130)

deeppavlov · Feb 25, 2020 · 3cb56cd · 3cb56cd
1 parent 69d5ee4
commit 3cb56cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeppavlov/metrics/bleu.py b/deeppavlov/metrics/bleu.py
@@ -78,4 +78,4 @@ def per_item_bleu(y_true, y_predicted):
 def per_item_dialog_bleu(y_true, y_predicted):
     y_true = (y['text'] for dialog in y_true for y in dialog)
     return corpus_bleu([[y_t.lower().split()] for y_t in y_true],
-                       [y_p.lower().split() for y_p in y_predicted])
+                       [y.lower().split() for y_p in y_predicted for y in y_p])