如何使用

需下载预训练模型ERNIE-health

  1. 安装依赖库
pip install -r requirements.txt
  1. 处理原始数据集
python preprocess.py
  1. 训练模型(含测试结果)
python main.py --mode='train'