MRC for DSD

BIBM2021 - A Knowledge-aware Machine Reading Comprehension Framework for Dialogue Symptom Diagnosis

Qiuck Start

0. Requirements

• python>=3.7
• torch==1.8.1
• transformers
• pandas
• sklearn
• numpy

1. Data Preprocess

The dataset used in this work can be obtained from CMDD

cd data
python preprocess.py

2. Training

python train.py

3. Predicting

python test.py

4. P, R and F1 calculate

python call_f1.py