diff --git a/convlab/policy/emoTUS/emoTUS-BertNLU-RuleDST-RulePolicy.json b/convlab/policy/emoTUS/emoTUS-BertNLU-RuleDST-RulePolicy.json new file mode 100644 index 0000000000000000000000000000000000000000..458b0628f0d387600696511f011ef540f0388380 --- /dev/null +++ b/convlab/policy/emoTUS/emoTUS-BertNLU-RuleDST-RulePolicy.json @@ -0,0 +1,55 @@ +{ + "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": { + "emoTUS": { + "class_path": "convlab.policy.emoTUS.emoTUS.UserPolicy", + "ini_params": { + "model_checkpoint": "convlab/policy/emoTUS/unify/experiments/EmoUS_emowoz+dialmage_0_1", + "use_sentiment": false, + "add_persona": true, + "sample": false + } + } + }, + "usr_nlg": {} +} \ No newline at end of file