PTM-based-E2E

Datasets

• MultiWOZ 2.0\2.1\2.2

Models

• UBAR: UBAR: Towards Fully End-to-End Task-Oriented Dialog System with GPT-2
• PPTOD: Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue System
• DAMD: Task-Oriented Dialog Systems that Consider Multiple Appropriate Responses under the Same Context

Data Preprocess

MultiWOZ

2.0: fast_dialog/data/dataset/: ./MultiWOZ2.0_preparation.sh
2.1: fast_dialog/data/dataset/: ./MultiWOZ2.1_preparation.sh
2.2: fast_dialog/data/dataset/: ./MultiWOZ2.2_preparation.sh

Training

PPTOD

PPTOD_small: fast_dialog/model/pptod/: ./pptod_small_train_full_training.sh
PPTOD_base: fast_dialog/model/pptod/: ./pptod_base_train_full_training.sh
PPTOD_large: fast_dialog/model/pptod/: ./pptod_large_train_full_training.sh

UBAR

UBAR: fast_dialog/model/ubar/: ./train_ubar.sh