KTGF

A Knowledge-enhanced Two-stage Generative Framework for Medical Dialogue Information Extraction, Machine Intelligence Research (MIR).

Requirement

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

Name		Name	Last commit message	Last commit date
Latest commit History 18 Commits
data_utils		data_utils
dataset		dataset
utils		utils
LICENSE		LICENSE
README.md		README.md
evaluate.py		evaluate.py
generate_CMDD.sh		generate_CMDD.sh
generate_stage1.py		generate_stage1.py
generate_stage1.sh		generate_stage1.sh
generate_stage2.py		generate_stage2.py
generate_stage2.sh		generate_stage2.sh
preprocess_data.py		preprocess_data.py
preprocess_data.sh		preprocess_data.sh
train.py		train.py
train.sh		train.sh