diff --git a/convlab/policy/emoUS/dialogue_collector.py b/convlab/policy/emoUS/dialogue_collector.py index 3422bed133a98d21775488db3c33f76ffdbba773..1976c0cd54bf5d3438622bb28f6d6a627ca03b7b 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 ec1b8e4465c73f9737c7fada8fcaaae89e933784..cff2780bc15a1d2dbd70d85a7cb7e88b4dfd81c3 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