├── .gitignore ├── LICENSE ├── README.md ├── data ├── DuRecDial2_en │ └── Readme.txt └── DuRecDial_zh │ └── Readme.txt ├── eval ├── eval_dialogue.py └── eval_planning.py ├── figure ├── constrained-decoding.png ├── overview-trip.png ├── plan-controlled-gen.png └── prompt-based-gen.png ├── main_dial.py ├── main_plan.py ├── model ├── generation_beam_constraints.py ├── generation_beam_search.py ├── generation_utils.py ├── model_trip.py ├── model_trip_dial.py └── nn_decoder.py ├── requirements.txt └── utils ├── data_collator.py ├── data_utils.py ├── dataset.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/README.md -------------------------------------------------------------------------------- /data/DuRecDial2_en/Readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/data/DuRecDial2_en/Readme.txt -------------------------------------------------------------------------------- /data/DuRecDial_zh/Readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/data/DuRecDial_zh/Readme.txt -------------------------------------------------------------------------------- /eval/eval_dialogue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/eval/eval_dialogue.py -------------------------------------------------------------------------------- /eval/eval_planning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/eval/eval_planning.py -------------------------------------------------------------------------------- /figure/constrained-decoding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/figure/constrained-decoding.png -------------------------------------------------------------------------------- /figure/overview-trip.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/figure/overview-trip.png -------------------------------------------------------------------------------- /figure/plan-controlled-gen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/figure/plan-controlled-gen.png -------------------------------------------------------------------------------- /figure/prompt-based-gen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/figure/prompt-based-gen.png -------------------------------------------------------------------------------- /main_dial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/main_dial.py -------------------------------------------------------------------------------- /main_plan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/main_plan.py -------------------------------------------------------------------------------- /model/generation_beam_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/model/generation_beam_constraints.py -------------------------------------------------------------------------------- /model/generation_beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/model/generation_beam_search.py -------------------------------------------------------------------------------- /model/generation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/model/generation_utils.py -------------------------------------------------------------------------------- /model/model_trip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/model/model_trip.py -------------------------------------------------------------------------------- /model/model_trip_dial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/model/model_trip_dial.py -------------------------------------------------------------------------------- /model/nn_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/model/nn_decoder.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/data_collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/utils/data_collator.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iwangjian/TRIP/HEAD/utils/trainer.py --------------------------------------------------------------------------------