MedLane

---

Setup

pip install -r requirements.txt

How to run

train on the entire dataset

python run.py \
    --data_path ./data \
    --train_file train(12809).txt \
    --test_file  test(2030).txt \

train on the sample dataset

python run.py \
    --data_path ./data \
    --train_file train(sample).txt \
    --test_file  test(sample).txt \

skip step1's training process and load a pretrained model

python run.py \
    --bert_path $BERT_PATH$ \
    --skip_step1

load a pretrained model for step2 and step3

python run.py \
    --para_path $PARA_PATH$ \