资讯 文档
技术能力
语音技术
文字识别
人脸与人体
图像技术
语言与知识
视频技术

进阶任务:使用ERNIE-tiny进行文本分类

使用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文件中。
上一篇
开始训练和预测(ERNIE百亿&15亿模型)
下一篇
法律领域多标签文本分类任务