xf-knowledgeClassify2021-top2
2021科大讯飞试题标签预测挑战赛亚军方案
Environment
环境：python3.6 + pytorch1.7
基于pytorch1.7环境，载入相关的安装包: "pip install -r requirements.txt"
执行方法：
- 模型训练：sh train.sh
- 模型预测：sh test.sh
模型加载(可选)
如果不想自己训练模型，可加载我训练好的模型。
将百度网盘中的“save_model”文件夹存放于主目录下；
（链接:https://pan.baidu.com/s/1nr6nsB5Qsm32MrMbaQUdww 密码:objk）
两点说明：
- 本方案只使用官方提供的数据集，未使用任何外部数据；
- 本方案的预训练模型为roberta_chinese_wwm_ext, 可在huggingface中进行下载。