Java Deep Learning Cookbook

How it works...

In step 1, we have used MnistDataSetIterator to extract and load MNIST data in one place. DL4J comes with this specialized iterator to load MNIST data without having to worry about downloading the data on your own. You might notice that MNIST data on the official website follows the ubyte format. This is certainly not the desired format, and we need to extract all the images separately to load them properly on the neural network.

Therefore, it is very convenient to have an MNIST iterator implementation such as MnistDataSetIterator in DL4J. It simplifies the typical task of handling MNIST data in the ubyte format. MNIST data has a total of 60,000 training digits, 10,000 test digits, and 10 labels. Digit images have a dimension ...