Implementing a simple CNN

In this recipe, we will develop a four-layer convolutional neural network to improve upon our accuracy in predicting MNIST digits. The first two convolution layers will each be composed of convolution-ReLU-Max Pool operations, and the final two layers will be fully connected layers.

Get TensorFlow Machine Learning Cookbook - Second Edition now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.