tatk是开源的面向任务型对话工具箱,由清华大学THU-coai开发。
提供了端到端模型。
tatk是dialtk的一部分。
We provide following models:
- NLU: SVMNLU, BERTNLU
- DST: rule, MDBT
- Policy: rule, Imitation, REINFORCE, PPO, MDRG
- Simulator policy: Agenda, VHUS
- NLG: Template, SCLSTM
- End2End: Sequicity, RNN_rollout
支持数据集:Multiwoz、Camrest、Dealornot
git clone https://github.com/thu-coai/tatk.git
huihui@192 git % source ~/anaconda3/bin/activate
(base) huihui@192 git % conda create --name env_tatk python=3.7
(base) huihui@192 git % conda activate env_tatk
(env_tatk) huihui@192 git % pip install jupyter
python setup.py build & install
(env_tatk) huihui@192 git % pip install jupyter
也可以pip安装pip install tatk
source ~/anaconda3/bin/activate
conda activate env_tatk
该仓库有演示demo