0
点赞
收藏
分享

微信扫一扫

表格多列情况下,loading不显示问题

模型训练如何实现自动化

1、目标

把对文本分类【体育,教育,娱乐,经济,文学,政治】的模型实现自动化训练。

分类器:贝叶斯、支持向量机、梯度提升、随机森林、逻辑斯蒂回归

通过预设的准确率期望值与训练轮数,获取最优的分类器模型,当模型评估的准确率大于期望值时即停止训练,保存模型、评估报告、特征与对应的训练数据主键(id)。

2、流程

1、设置模型初始准确率期望值与训练轮数
2、多个分类器多线程训练
3、获取特征并随机划分训练集与测试集
4、保存模型、评估报告、特征与对应的训练数据主键(id)

3、输出效果

在这里插入图片描述

在这里插入图片描述

4、实现代码

举报

相关推荐

0 条评论