使用NEAT进行图像分类

0. 前言

为了真正发挥 NEAT 能力,并了解其限制,本节将使用 MNIST 手写数字数据集执行图像分类任务,使用 NEATMNIST 数据集进行分类。

1. 数据集加载

加载 MNIST 数据集,我们只使用数据集的训练数据部分来评估基因组在批样本上的适应度。数据加载后,进行归一化,然后显示示例数字,本节使用完整数据集中的 10 个类别:

import numpy as np
import tensorflow as tf
import matplotlib.pyplot