A classifier (this is for supervised learning)
Interface for a layer of a neural network.
A Model is meant for predicting something from data.
This interface describes abstraction that uses provided model to convert INDArrays to some specific output
Param initializer for a layer
Trainable: an interface common to Layers and GraphVertices that have trainable parameters
Simple interface for the training configuration (updater, L1/L2 values, etc) for trainable layers/vertices.
Update the model
Type of forward pass to do.
MaskState: specifies whether a mask should be applied or not.
Optimization algorithm to use
Copyright © 2020. All rights reserved.