Skip to content

Latest commit

 

History

History
40 lines (28 loc) · 547 Bytes

README.md

File metadata and controls

40 lines (28 loc) · 547 Bytes

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