Home

Awesome

Task-Optimized Adatper for an End-to-End Task Oriented Dialog

Prerequisite

Install Requirements

pip install -r requirements.txt

wandb setup

pip install wandb
wandb login
wandb init

Download & Preprocess Data

MultiWOZ 2.1

cd data/multiwoz21
bash data_preparation.sh

MultiWOZ 2.2

cd data/multiwoz22
bash data_preparation.sh

NLU task

cd data/banking77
bash banking77_preparation.sh

cd ../clinc150
bash clinc150_preparation.sh

cd ../hwu64
bash ../hwu64_preparation.sh

Download Pretrained Model

cd checkpoints

wget https://pptod.s3.amazonaws.com/Pretrain/small.zip
unzip small.zip
rm small.zip

wget https://pptod.s3.amazonaws.com/Pretrain/base.zip
unzip base.zip
rm base.zip

Train & Eval of E2E TOD

small

cd E2E_TOD
bash small_run_21.sh
bash small_run_22.sh

base

cd E2E_TOD
bash base_run_21.sh
bash base_run_22.sh

Train & Eval of NLU(IC)

cd IC
bash run.sh