tatkを使用してダイアログエージェントを構築する

845 ワード

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があります