Skip to Main Content
Hands-On Deep Learning with Apache Spark
book

Hands-On Deep Learning with Apache Spark

by Guglielmo Iozzia
January 2019
Intermediate to advanced content levelIntermediate to advanced
322 pages
7h 29m
English
Packt Publishing
Content preview from Hands-On Deep Learning with Apache Spark

CNN distributed training with Spark and DL4J

Let's get back to the example that has been presented in Chapter 5, Convolutional Neural Networks, Hands-on CNN with Spark, about handwritten digits image classification on the MNIST dataset. For convenience, here's a reminder of the network configuration used there:

val channels = 1 val outputNum = 10 val conf = new NeuralNetConfiguration.Builder()     .seed(seed)     .iterations(iterations)     .regularization(true)     .l2(0.0005)     .learningRate(.01)     .weightInit(WeightInit.XAVIER)     .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)     .updater(Updater.NESTEROVS)     .momentum(0.9)     .list     .layer(0, new ConvolutionLayer.Builder(5, 5)         .nIn(channels)         .stride(1, 1)         .nOut(20) .activation(Activation.IDENTITY) ...
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.
Start your free trial

You might also like

Next-Generation Machine Learning with Spark: Covers XGBoost, LightGBM, Spark NLP, Distributed Deep Learning with Keras, and More

Next-Generation Machine Learning with Spark: Covers XGBoost, LightGBM, Spark NLP, Distributed Deep Learning with Keras, and More

Butch Quinto
Apache Spark Deep Learning Cookbook

Apache Spark Deep Learning Cookbook

Ahmed Sherif, Amrith Ravindra, Michal Malohlava, Adnan Masood

Publisher Resources

ISBN: 9781788994613Supplemental Content