Saving and loading trained neural network models

Training the neural network over and over to perform an evaluation is not a good idea since training is a very costly operation. This is why model persistence is important in distributed systems as well.

In this recipe, we will persist the distributed neural network models to disk and load them for further use.

How to do it...

1. Save the distributed neural network model using ModelSerializer:

MultiLayerNetwork model = sparkModel.getNetwork();
 File file = new File("MySparkMultiLayerNetwork.bin");
 ModelSerializer.writeModel(model,file, saveUpdater);

2. Save the distributed neural network model using save():

MultiLayerNetwork model = sparkModel.getNetwork();
  File locationToSave...