Skip to content
Snippets Groups Projects
Commit a169c0af authored by Hsien-Chin Lin's avatar Hsien-Chin Lin
Browse files

Merge branch 'genTUS_v2' of gitlab.cs.uni-duesseldorf.de:dsml/convlab/ConvLab3 into genTUS_v2

parents ce8b6ba8 6b326374
No related branches found
No related tags found
No related merge requests found
...@@ -620,6 +620,7 @@ class UserPolicy(Policy): ...@@ -620,6 +620,7 @@ class UserPolicy(Policy):
else: else:
mode = "max" mode = "max"
response = self.policy.predict(sys_act, mode) response = self.policy.predict(sys_act, mode)
print(response)
self.semantic_action = self.policy.semantic_action self.semantic_action = self.policy.semantic_action
return response return response
......
...@@ -8,7 +8,7 @@ ...@@ -8,7 +8,7 @@
"epoch": 100, "epoch": 100,
"eval_frequency": 5, "eval_frequency": 5,
"process_num": 1, "process_num": 1,
"num_eval_dialogues": 200, "num_eval_dialogues": 20,
"sys_semantic_to_usr": false "sys_semantic_to_usr": false
}, },
"vectorizer_sys": { "vectorizer_sys": {
......
...@@ -8,7 +8,7 @@ ...@@ -8,7 +8,7 @@
"epoch": 100, "epoch": 100,
"eval_frequency": 5, "eval_frequency": 5,
"process_num": 1, "process_num": 1,
"num_eval_dialogues": 200, "num_eval_dialogues": 20,
"sys_semantic_to_usr": false "sys_semantic_to_usr": false
}, },
"vectorizer_sys": { "vectorizer_sys": {
...@@ -49,7 +49,7 @@ ...@@ -49,7 +49,7 @@
"mode": "language", "mode": "language",
"only_action": false, "only_action": false,
"use_sentiment": true, "use_sentiment": true,
"sample": true "sample": false
} }
} }
}, },
......
{
"model": {
"load_path": "convlab/policy/ppo/pretrained_models/mle",
"pretrained_load_path": "",
"use_pretrained_initialisation": false,
"batchsz": 200,
"seed": 0,
"epoch": 100,
"eval_frequency": 5,
"process_num": 1,
"num_eval_dialogues": 20,
"sys_semantic_to_usr": false
},
"vectorizer_sys": {
"uncertainty_vector_mul": {
"class_path": "convlab.policy.vector.vector_binary.VectorBinary",
"ini_params": {
"use_masking": true,
"manually_add_entity_names": true,
"seed": 0
}
}
},
"nlu_sys": {
"BertNLU": {
"class_path": "convlab.nlu.jointBERT.unified_datasets.BERTNLU",
"ini_params": {
"mode": "all",
"config_file": "multiwoz21_all.json",
"model_file": "https://huggingface.co/ConvLab/bert-base-nlu/resolve/main/bertnlu_unified_multiwoz21_all_context0.zip"
}
}
},
"dst_sys": {
"RuleDST": {
"class_path": "convlab.dst.rule.multiwoz.dst.RuleDST",
"ini_params": {}
}
},
"sys_nlg": {},
"nlu_usr": {},
"dst_usr": {},
"policy_usr": {
"GenTUS": {
"class_path": "convlab.policy.genTUS.stepGenTUS.UserPolicy",
"ini_params": {
"model_checkpoint": "convlab/policy/genTUS/unify/experiments/multiwoz21-exp",
"mode": "language",
"only_action": false
}
}
},
"usr_nlg": {}
}
{
"model": {
"load_path": "convlab/policy/ppo/pretrained_models/mle",
"pretrained_load_path": "",
"use_pretrained_initialisation": false,
"batchsz": 200,
"seed": 0,
"epoch": 100,
"eval_frequency": 5,
"process_num": 1,
"num_eval_dialogues": 20,
"sys_semantic_to_usr": false
},
"vectorizer_sys": {
"uncertainty_vector_mul": {
"class_path": "convlab.policy.vector.vector_binary.VectorBinary",
"ini_params": {
"use_masking": true,
"manually_add_entity_names": true,
"seed": 0
}
}
},
"nlu_sys": {
"BertNLU": {
"class_path": "convlab.nlu.jointBERT.unified_datasets.BERTNLU",
"ini_params": {
"mode": "all",
"config_file": "multiwoz21_all.json",
"model_file": "https://huggingface.co/ConvLab/bert-base-nlu/resolve/main/bertnlu_unified_multiwoz21_all_context0.zip"
}
}
},
"dst_sys": {
"RuleDST": {
"class_path": "convlab.dst.rule.multiwoz.dst.RuleDST",
"ini_params": {}
}
},
"sys_nlg": {},
"nlu_usr": {},
"dst_usr": {},
"policy_usr": {
"RulePolicy": {
"class_path": "convlab.policy.emoTUS.emoTUS.UserPolicy",
"ini_params": {
"model_checkpoint": "convlab/policy/emoTUS/unify/experiments/emowoz+dialmage_0_1/23-01-11-15-17",
"character": "usr",
"mode": "language",
"only_action": false,
"use_sentiment": true,
"sample": false
}
}
},
"usr_nlg": {}
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment