用于 MNIST 数据的 LeNet
您可以按照 Jupyter 笔记本中的代码ch-09a_CNN_MNIST_TF_and_Keras
。
准备 MNIST 数据到测试和训练集:
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets(os.path.join('.','mnist'), one_hot=True)
X_train = mnist.train.images
X_test = mnist.test.images
Y_train = mnist.train.labels
Y_test = mnist.test.labels