模型训练如何实现自动化
1、目标
把对文本分类【体育,教育,娱乐,经济,文学,政治】的模型实现自动化训练。
分类器:贝叶斯、支持向量机、梯度提升、随机森林、逻辑斯蒂回归
通过预设的准确率期望值与训练轮数,获取最优的分类器模型,当模型评估的准确率大于期望值时即停止训练,保存模型、评估报告、特征与对应的训练数据主键(id)。
2、流程
1、设置模型初始准确率期望值与训练轮数
2、多个分类器多线程训练
3、获取特征并随机划分训练集与测试集
4、保存模型、评估报告、特征与对应的训练数据主键(id)