Interface | Description |
---|---|
Classifier |
A classifier (this is for supervised learning)
|
Layer |
Interface for a layer of a neural network.
|
Model |
A Model is meant for predicting something from data.
|
ParamInitializer |
Param initializer for a layer
|
Updater |
Update the model
|
Enum | Description |
---|---|
Layer.TrainingMode | |
Layer.Type | |
MaskState |
MaskState: specifies whether a mask should be applied or not.
|
OptimizationAlgorithm |
Optimization algorithm to use
|