fix the algo and eval

Change-Id: If163c03c1acccf4c6303815f6fdad4589a71385f
wxc68762040 · Nov 10, 2017 · 58aafa9 · 58aafa9
1 parent 3252b9f
commit 58aafa9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/rc_model.py b/tensorflow/rc_model.py
@@ -333,7 +333,7 @@ def evaluate(self, eval_batches, result_dir=None, result_prefix=None, save_full_
         if len(ref_answers) > 0:
             pred_dict = {pred['query_id']: pred['answers'] for pred in pred_answers}
             ref_dict = {ref['query_id']: ref['answers'] for ref in ref_answers}
-            bleu_rouge = compute_bleu_rouge(pred_answers, ref_answers)
+            bleu_rouge = compute_bleu_rouge(pred_dict, ref_dict)
         else:
             bleu_rouge = None
         return ave_loss, bleu_rouge