使用数据增强方法提升模型性能


使用数据增强来提升模型的性能

1、导入模型

2、定义加载函数

3、定义批量加载函数

4、加载数据

把数据转换为dataset格式

5、定义数据增强方法

6、构建模型

生成实例

查看模型详细结构

7、编译模型

8、训练模型

为便于比较,这里先不使用数据增强方法

9、查看运行结果

运行结果

10、使用数据增强方法

11、查看使用数据增强的运行结果

运行结果

12、结果分析

从不使用数据增强与使用数据增强方法的结果可以看出,使用数据增强方法后,模型性能有提升(未使用数据增强的验证精度为71%,使用数据增强方法后,验证精度提升到74%),而且模型的泛化能力也有提高(使用数据增强方法后,训练与验证精度曲线靠得较近)。

发表评论