From 2ae7cb7dbf5ef75bd813a1371830a30f42cd6d6b Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Mon, 20 Feb 2023 17:37:53 +0100
Subject: [PATCH] wip
---
convlab/policy/emoTUS/evaluate.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py
index 900c9e80..3fbd348a 100644
--- a/convlab/policy/emoTUS/evaluate.py
+++ b/convlab/policy/emoTUS/evaluate.py
@@ -136,6 +136,8 @@ class Evaluator:
for dialog in tqdm(in_file['dialog']):
for x in dialog:
+ if x not in self.r:
+ self.r[x] = []
self.r[x].append(dialog[x])
def _transform_result(self):
--
GitLab