背景:官网的应用案例教程力求简洁,主要描述模型结构和训练验证推理过程,其他部分可尽量简洁或者略去。
1 添加mindrecord之前的数据处理过程代码:
![](https://i-blog.csdnimg.cn/blog_migrate/91d57d1258a8d365def077f573417780.png)
STMJ_dataset类会用到以上3个类和1个方法。
2 训练函数的代码是这样的
![](https://i-blog.csdnimg.cn/blog_migrate/2a5cfcda74e86991689a8111c81bb310.png)
3 添加mindrecord为中间数据的代码
![](https://i-blog.csdnimg.cn/blog_migrate/fdc615fb65899ed4b233a1f99f1c60dd.png)
只需要在数据读取过程中添加一行代码:data_train.save("STMJ_train_dataset.mindrecord")。
在运行过程中直接读取mindrecord类型数据,获得的效果基本一样,不会对模型有任何的影响,效果如下:
![](https://i-blog.csdnimg.cn/blog_migrate/329aa7b3572612da918464afbe0a1bf3.png)
![](https://i-blog.csdnimg.cn/blog_migrate/dfc988731f76594fdebc7f731810da17.png)
4 结果分析
这样操作的结果就是能够省去“1.添加mindrecord之前的数据处理过程代码过程”的5个方法,为图文案例节省去很大的篇幅。
需要注意的是:“3.添加mindrecord为中间数据的代码”这部分的第二张图里面框出来部分,位置有交换,实际在应用过程中需要调换位置赋值。