diff --git a/convlab/policy/emoUS/emoUS.py b/convlab/policy/emoUS/emoUS.py index 165195ccab71cfbee5c4712f042a44e353556685..2f0c24b78f95d273b0d0855134a429a736fdd0c0 100644 --- a/convlab/policy/emoUS/emoUS.py +++ b/convlab/policy/emoUS/emoUS.py @@ -464,7 +464,7 @@ if __name__ == "__main__": import time use_sentiment, emotion_mid = False, False - set_seed(0) + set_seed(100) # Test semantic level behaviour usr_policy = UserPolicy( # model_checkpoint, # default location = convlab/policy/emoUS/unify/default/EmoUS_default @@ -476,15 +476,15 @@ if __name__ == "__main__": usr_nlu = None # BERTNLU() usr = PipelineAgent(usr_nlu, None, usr_policy, None, name='user') usr.init_session() + usr.init_session() print(usr.policy.get_goal()) start = time.time() - print(usr.response([]), usr.policy.get_emotion()) + # print(usr.policy.policy.goal.status) - print(usr.response([["inform", "restaurant", "area", "centre"], - ["request", "restaurant", "food", "?"]]), + print(usr.response([['inform', 'train', 'day', 'saturday']]), usr.policy.get_emotion()) # print(usr.policy.policy.goal.status) - print(usr.response([["request", "restaurant", "price range", "?"]]), + print(usr.response([]), usr.policy.get_emotion()) end = time.time() print("-"*50) diff --git a/convlab/policy/emoUS/emotion_eval.py b/convlab/policy/emoUS/emotion_eval.py index be0c383c3d0d80a5701a36a1779b5cd07c60da2b..ee9fc61650f7162b1065bc7ca89499b25758c63c 100644 --- a/convlab/policy/emoUS/emotion_eval.py +++ b/convlab/policy/emoUS/emotion_eval.py @@ -11,7 +11,7 @@ from sklearn import metrics from tqdm import tqdm from convlab.nlg.evaluate import fine_SER -from convlab.policy.emoTUS.emoTUS import UserActionPolicy +from convlab.policy.emoUS.emoUS import UserActionPolicy sys.path.append(os.path.dirname(os.path.dirname( os.path.dirname(os.path.abspath(__file__))))) @@ -72,13 +72,13 @@ class Evaluator: self.emotion_list = [] - for emotion in json.load(open("convlab/policy/emoTUS/emotion.json")): + for emotion in json.load(open("convlab/policy/emoUS/emotion.json")): self.emotion_list.append(emotion) self.r[f"{emotion}_acts"] = [] self.r[f"{emotion}_utts"] = [] sent2emo = json.load( - open("convlab/policy/emoTUS/sent2emo.json")) + open("convlab/policy/emoUS/sent2emo.json")) self.emo2sent = {} for sent, emotions in sent2emo.items(): for emo in emotions: