From 32c3731a82b94e7a6b4cefec153dbe07a37648f0 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Tue, 24 Jan 2023 00:13:56 +0100
Subject: [PATCH] wip
---
convlab/policy/emoTUS/evaluate.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py
index 86af813d..2204c26d 100644
--- a/convlab/policy/emoTUS/evaluate.py
+++ b/convlab/policy/emoTUS/evaluate.py
@@ -85,7 +85,7 @@ class Evaluator:
emotion_mode = "normal"
in_file = json.load(open(f_eval))
- for dialog in tqdm(in_file['dialog'][:2]):
+ for dialog in tqdm(in_file['dialog']):
inputs = dialog["in"]
labels = self.usr._parse_output(dialog["out"])
--
GitLab