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

98 篇文章 110 订阅
70 篇文章 34 订阅

项目介绍

本文将基于 Tensorflow2.0 使用遗传算法对网络结构寻优来对 cifar10 数据集进行分类。此项目共包含两个文件,一个是用于导入数据集、构建网络以及训练网络的文件 main.py,另一个是实现遗传算法的文件 GA.py。

在构建网络时,我们将采用模块堆叠的方式,仿照 MobileNet V2 中的模块结构;在训练网络时,我们将学习率和优化器类型视为可变参数与网络结构共同放入遗传算法进行寻优。

main.py

1、导入需要的库

import tensorflow as tf
import numpy as np
import os

2、导入数据集

我们将这一操作封装到 load() 函数中,方便 GA.py 文件的调用。

def load(
  • 5
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

cofisher

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值