Running a CNN for learning MNIST with DeepLearning4j over Spark

In this recipe, we'll see how to run a CNN for classifying the iris dataset.

Getting ready

To step through this recipe, you will need a running Spark cluster either in pseudo distributed mode or in one of the distributed modes, that is, standalone, YARN, or Mesos. Also, get familiar with ND4S, that is, n-dimensional arrays for Scala (Scala bindings for ND4J). ND4J and ND4S are scientific computing libraries for the JVM. Please visit http://nd4j.org/ for details. The prerequisites to be installed are Java 7, IntelliJ, and the Maven or SBT build tool.

How to do it…

  1. The MNIST database is a large set of handwritten digits used to train neural networks and other algorithms in image recognition. ...

Get Apache Spark for Data Science Cookbook 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.