🏥 📖 Mandarin medical dialogue analysis implemented in PyTorch.
This is a repository for analyzing mandarin medical dialgoues, including risk assessment and multiple-choice questing answering. The codes are designed for the Medical Dialog Analysis Competition. To reproduce our results, please refer to the reproduce section.
- Python 3.6+
pip install -r requirement.txtvim aidea-web.tw_cookies.txt # put log-in cookies.txt here
./download.shbash reproduce.sh config/risk_test.yaml config/qa_test.yamlSee report.pdf
vim config/mlm.yaml # Set train path and pretrained model.
bash train_mlm.shvim config/risk.yaml # Put your pretrained MLM here. Set train path and test path.
bash train_risk.shpython src/train_risk.py --config config/risk.yaml --mode testbash run_risk_tfidf.shSee report.pdf
bash run_qa_rule_base.sh