背景:官网的应用案例教程力求简洁,主要描述模型结构和训练验证推理过程,其他部分可尽量简洁或者略去。
1 添加mindrecord之前的数据处理过程代码:

STMJ_dataset类会用到以上3个类和1个方法。
2 训练函数的代码是这样的

3 添加mindrecord为中间数据的代码

只需要在数据读取过程中添加一行代码:data_train.save("STMJ_train_dataset.mindrecord")。
在运行过程中直接读取mindrecord类型数据,获得的效果基本一样,不会对模型有任何的影响,效果如下:


4 结果分析
这样操作的结果就是能够省去“1.添加mindrecord之前的数据处理过程代码过程”的5个方法,为图文案例节省去很大的篇幅。
需要注意的是:“3.添加mindrecord为中间数据的代码”这部分的第二张图里面框出来部分,位置有交换,实际在应用过程中需要调换位置赋值。