tatkを使用してダイアログエージェントを構築する
845 ワード
tatkはオープンソースのタスク向け対話ツールボックスで、清華大学THU-coaiが開発した.
エンドツーエンドモデルが提供されます.
tatkはdialtkの一部です.
サポートデータセット:Multiwoz、Camrest、Dealornot
この倉庫にはデモdemoがあります
エンドツーエンドモデルが提供されます.
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があります