/fast_dialog

A benchmark for Task Oriented Dialog

Primary LanguagePython

PTM-based-E2E

Datasets

  • MultiWOZ 2.0\2.1\2.2

Models

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