├── .gitignore ├── LICENSE ├── README.md ├── data ├── DuRecDial2.0 │ └── Readme.txt └── TGConv │ └── Readme.txt ├── eval ├── eval_dialog_durecdial.py ├── eval_dialog_tgconv.py ├── eval_dialog_tgconv_selfplay.py └── eval_planning.py ├── figure └── overview.png ├── main_dialog.py ├── main_planning.py ├── model ├── generation_utils.py ├── model_color.py ├── model_dialog.py └── modeling_bart.py ├── requirements.txt ├── scripts ├── durecdial_dialog_test.sh ├── durecdial_dialog_train.sh ├── durecdial_planning_infer_planner.sh ├── durecdial_planning_train_bridge.sh ├── durecdial_planning_train_planner.sh ├── tgconv_dialog_test_selfplay.sh ├── tgconv_dialog_train.sh ├── tgconv_planning_infer_planner.sh ├── tgconv_planning_train_bridge.sh └── tgconv_planning_train_planner.sh ├── train ├── trainer_bridge.py ├── trainer_dialog.py └── trainer_planner.py └── utils ├── data_collator.py ├── data_utils.py ├── dataset_base.py ├── dataset_durecdial.py └── dataset_tgconv.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/README.md -------------------------------------------------------------------------------- /data/DuRecDial2.0/Readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/data/DuRecDial2.0/Readme.txt -------------------------------------------------------------------------------- /data/TGConv/Readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/data/TGConv/Readme.txt -------------------------------------------------------------------------------- /eval/eval_dialog_durecdial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/eval/eval_dialog_durecdial.py -------------------------------------------------------------------------------- /eval/eval_dialog_tgconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/eval/eval_dialog_tgconv.py -------------------------------------------------------------------------------- /eval/eval_dialog_tgconv_selfplay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/eval/eval_dialog_tgconv_selfplay.py -------------------------------------------------------------------------------- /eval/eval_planning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/eval/eval_planning.py -------------------------------------------------------------------------------- /figure/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/figure/overview.png -------------------------------------------------------------------------------- /main_dialog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/main_dialog.py -------------------------------------------------------------------------------- /main_planning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/main_planning.py -------------------------------------------------------------------------------- /model/generation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/model/generation_utils.py -------------------------------------------------------------------------------- /model/model_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/model/model_color.py -------------------------------------------------------------------------------- /model/model_dialog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/model/model_dialog.py -------------------------------------------------------------------------------- /model/modeling_bart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/model/modeling_bart.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/durecdial_dialog_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/durecdial_dialog_test.sh -------------------------------------------------------------------------------- /scripts/durecdial_dialog_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/durecdial_dialog_train.sh -------------------------------------------------------------------------------- /scripts/durecdial_planning_infer_planner.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/durecdial_planning_infer_planner.sh -------------------------------------------------------------------------------- /scripts/durecdial_planning_train_bridge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/durecdial_planning_train_bridge.sh -------------------------------------------------------------------------------- /scripts/durecdial_planning_train_planner.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/durecdial_planning_train_planner.sh -------------------------------------------------------------------------------- /scripts/tgconv_dialog_test_selfplay.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/tgconv_dialog_test_selfplay.sh -------------------------------------------------------------------------------- /scripts/tgconv_dialog_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/tgconv_dialog_train.sh -------------------------------------------------------------------------------- /scripts/tgconv_planning_infer_planner.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/tgconv_planning_infer_planner.sh -------------------------------------------------------------------------------- /scripts/tgconv_planning_train_bridge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/tgconv_planning_train_bridge.sh -------------------------------------------------------------------------------- /scripts/tgconv_planning_train_planner.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/scripts/tgconv_planning_train_planner.sh -------------------------------------------------------------------------------- /train/trainer_bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/train/trainer_bridge.py -------------------------------------------------------------------------------- /train/trainer_dialog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/train/trainer_dialog.py -------------------------------------------------------------------------------- /train/trainer_planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/train/trainer_planner.py -------------------------------------------------------------------------------- /utils/data_collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/utils/data_collator.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/dataset_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/utils/dataset_base.py -------------------------------------------------------------------------------- /utils/dataset_durecdial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/utils/dataset_durecdial.py -------------------------------------------------------------------------------- /utils/dataset_tgconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/Color4Dial/HEAD/utils/dataset_tgconv.py --------------------------------------------------------------------------------