使用tatk构建对话代理

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

原文地址:https://www.cnblogs.com/xuehuiping/p/12986498.html