fix bug in test-mode evaluation (wrong key for transfer tasks)

whuhxb · Jul 1, 2021 · cc0a168 · cc0a168
1 parent 0f7e8e8
commit cc0a168
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluation.py b/evaluation.py
@@ -193,7 +193,7 @@ def batcher(params, batch, max_length=None):
         for task in ['MR', 'CR', 'SUBJ', 'MPQA', 'SST2', 'TREC', 'MRPC']:
             task_names.append(task)
             if task in results:
-                scores.append("%.2f" % (results[task]['devacc']))    
+                scores.append("%.2f" % (results[task]['acc']))    
             else:
                 scores.append("0.00")
         task_names.append("Avg.")