Tensorflow2.0 用遗传算法进行网络架构搜索

文章目录

  • 项目介绍
  • 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 中的模块结构

你可能感兴趣的:(Tensorflow,2.0,深度学习,优化算法优化神经网络,深度学习,tensorflow,python,遗传算法,网络结构)