From 985e51b69fc9937f22c06627d3ce6d4a13d18c25 Mon Sep 17 00:00:00 2001
From: truthless11 <truthless11@gmail.com>
Date: Wed, 23 Sep 2020 16:59:48 +0800
Subject: [PATCH] fix policy evaluation
---
convlab2/policy/evaluate.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/convlab2/policy/evaluate.py b/convlab2/policy/evaluate.py
index 8085bf4..de0a06f 100755
--- a/convlab2/policy/evaluate.py
+++ b/convlab2/policy/evaluate.py
@@ -222,8 +222,7 @@ def evaluate(dataset_name, model_name, load_path, calculate_reward=True):
for key in sess.evaluator.goal:
if key not in task_success:
task_success[key] = []
- else:
- task_success[key].append(task_succ)
+ task_success[key].append(task_succ)
task_success['All'].append(task_succ)
for key in task_success:
--
GitLab