org.deeplearning4j.scalnet.models
Sequential
Related Doc:
package models
Compile neural net architecture.
Compile neural net architecture. Call immediately before training.
loss function to use
optimization algorithm to use
Fit neural net to data.
Fit neural net to data.
iterator over data set
number of epochs to train
callbacks for monitoring training
Use neural net to make prediction on input x
Use neural net to make prediction on input x
input represented as INDArray
Use neural net to make prediction on input x.
Class for keras-style simple sequential neural net architectures with one input node and one output node for each node in computational graph.
Wraps DL4J MultiLayerNetwork. Enforces keras model construction pattern: preprocessing (reshaping) layers should be explicitly provided by the user, while last layer is treated implicitly as an output layer.