机器学习实战:基于Sophon平台的机器学习理论与实践
上QQ阅读APP看书,第一时间看更新

1.2.2 流程搭建

参考1.1.4节建立模型训练与预测的流程(如图1-4所示)。其中String Index和set role属于数据预处理阶段,分别对label设置index和label角色。然后进行数据切分,将数据切分为训练集和验证集,比例为8/2。接下来,利用人工神经网络进行深度学习训练。图1-5展示了深度学习各层的配置。

图1-4 mnist数据集:流程

图1-5 mnist数据集:深度学习各层配置

最终模型训练完毕,可以应用模型并在验证集上验证模型性能。图1-6给出了混淆矩阵和召回率。关于Sophon机器学习平台的详细介绍请参见附录A。

图1-6 mnist数据集:模型性能