Java Deep Learning Cookbook

How to do it...

1. Create the ComputationGraph model from the previously-created model configuration:

ComputationGraphConfiguration configuration = builder.build();   ComputationGraph model = new ComputationGraph(configuration);

2. Load the iterator and train the model using the fit() method:

for(int i=0;i<epochs;i++){   model.fit(trainDataSetIterator); }

You can use the following approach as well:

model.fit(trainDataSetIterator,epochs);

We can then avoid using a for loop by directly specifying the epochs parameter in the fit() method.