A Knowledge-enhanced Two-stage Generative Framework for Medical Dialogue Information Extraction, Machine Intelligence Research (MIR).
- pytorch 1.10.0
- transformers 4.18.0
Download the Chinese T5 base and save the files to pretrained_models/t5-base-chinese-cluecorpussmall Folder.
sh preprocess_data.sh
sh train.sh
sh generate_stage1.sh
sh generate_stage2.sh