From bb9921c402f658315c44a20a42f8eaa78693d87b Mon Sep 17 00:00:00 2001 From: Hsien-Chin Lin <linh@hhu.de> Date: Thu, 13 Apr 2023 14:42:57 +0200 Subject: [PATCH] update --- convlab/policy/emoUS/dialogue_collector.py | 2 +- convlab/policy/emoUS/evaluate.py | 2 -- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/convlab/policy/emoUS/dialogue_collector.py b/convlab/policy/emoUS/dialogue_collector.py index 3422bed1..1976c0cd 100644 --- a/convlab/policy/emoUS/dialogue_collector.py +++ b/convlab/policy/emoUS/dialogue_collector.py @@ -95,7 +95,7 @@ if __name__ == "__main__": num_goals=args.num) data = {"config": json.load(open(args.config)), "conversation": conversation} - folder_name = os.path.join("convlab/policy/emoTUS", "conversation") + folder_name = os.path.join("convlab/policy/emoUS", "conversation") if not os.path.exists(folder_name): os.makedirs(folder_name) json.dump(data, diff --git a/convlab/policy/emoUS/evaluate.py b/convlab/policy/emoUS/evaluate.py index ec1b8e44..cff2780b 100644 --- a/convlab/policy/emoUS/evaluate.py +++ b/convlab/policy/emoUS/evaluate.py @@ -7,8 +7,6 @@ from datetime import datetime import matplotlib.pyplot as plt import torch from datasets import load_metric -# from convlab.policy.genTUS.pg.stepGenTUSagent import \ -# stepGenTUSPG as UserPolicy from sklearn import metrics from tqdm import tqdm -- GitLab