Running an RBM with DeepLearning4j over Spark

In this recipe, we'll see how to run a restricted Boltzmann machine 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 pre-requisites to be installed are Java 7, IntelliJ, and the Maven or SBT build tool.

How to do it…

  1. Start an application named RBMWithSpark. Initially, specify the following libraries in the build.sbt file:
     libraryDependencies ...

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.

Start your free trial