如何使用

需下载预训练模型ERNIE-health

1. 安装依赖库

pip install -r requirements.txt

2. 处理原始数据集

python preprocess.py

3. 训练模型(含测试结果)

python main.py --mode='train'