进阶任务:使用ERNIE-tiny进行文本分类
更新时间:2022-07-05
使用ERNIE_Tiny训练模型:
- 下载ERNIE_tiny模型:
# ernie_tiny_1.0 模型下载
# 进入models_hub目录
cd ../../models_hub/
# 运行下载脚本
sh download_ernie_tiny_1.0_ch.sh
- 通过加载预置json配置文件进行模型训练
python run_trainer.py --param_path ./examples/cls_ernie_tiny_1.0_fc_ch.json
预测我的第一个模型
- 开始预测(由于ernie_tiny已开源,故训练和预测都走不加密的策略)
python run_infer_ernie_model.py --param_path ./examples/cls_ernie_tiny_1.0_fc_ch_infer.json
- 预测运行的日志会自动保存在./output/predict_result.txt文件中。