文章目录
- 项目介绍
- main.py
-
- 1、导入需要的库
- 2、导入数据集
- 3、构造模型
- 4、训练模型
- GA.py
-
- 1、导入需要的库
- 2、设置遗传算法参数
- 3、导入数据集
- 4、适应度函数
- 5、选择函数
- 6、交叉函数
- 7、变异函数
- 8、初始化种群
- 9、迭代寻优
项目介绍
本文将基于 Tensorflow2.0 使用遗传算法对网络结构寻优来对 cifar10 数据集进行分类。此项目共包含两个文件，一个是用于导入数据集、构建网络以及训练网络的文件 main.py，另一个是实现遗传算法的文件 GA.py。
在构建网络时，我们将采用模块堆叠的方式，仿照 MobileNet V2 中的模块结构