diff --git a/README.md b/README.md index ab1741838c939b13fc580168544b9c5895cea9c6..3306c2345ea6c59e89820cdc15eb6dd011bf1cf1 100755 --- a/README.md +++ b/README.md @@ -9,7 +9,22 @@ The code of TUS is in `convlab2/policy/tus` and a rule-based DST of user is also ### Train the user simulator `python3 convlab2/policy/tus/multiwoz/train.py --user_config convlab2/policy/tus/multiwoz/exp/default.json` +One default configuration is placed in `convlab2/policy/tus/multiwoz/exp/default.json` + ### Train a dialogue policy with TUS +You can use it as a normal user simulator by `PipelineAgent`. For example, +```python +import json +from convlab2.dialog_agent.agent import PipelineAgent +from convlab2.dst.rule.multiwoz.usr_dst import UserRuleDST +from convlab2.policy.tus.multiwoz.TUS import UserPolicy + +dst_usr = UserRuleDST() +user_config = json.load(open(args.user_config)) +policy_usr = UserPolicy(user_config) +simulator = PipelineAgent(None, dst_usr, policy_usr, None, 'user') +``` + <!---citation--->