| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff | |
| org.nd4j.linalg.learning | |
| org.nd4j.linalg.learning.config |
| Modifier and Type | Method and Description |
|---|---|
TrainingConfig.Builder |
TrainingConfig.Builder.updater(IUpdater updater)
|
| Constructor and Description |
|---|
TrainingConfig(IUpdater updater,
List<Regularization> regularization,
boolean minimize,
List<String> dataSetFeatureMapping,
List<String> dataSetLabelMapping,
List<String> dataSetFeatureMaskMapping,
List<String> dataSetLabelMaskMapping,
List<String> lossVariables)
Create a training configuration suitable for training both single input/output and multi input/output networks.
See also the TrainingConfig.Builder for creating a TrainingConfig |
TrainingConfig(IUpdater updater,
List<Regularization> regularization,
boolean minimize,
List<String> dataSetFeatureMapping,
List<String> dataSetLabelMapping,
List<String> dataSetFeatureMaskMapping,
List<String> dataSetLabelMaskMapping,
List<String> lossVariables,
Map<String,List<IEvaluation>> trainEvaluations,
Map<String,Integer> trainEvaluationLabels,
Map<String,List<IEvaluation>> validationEvaluations,
Map<String,Integer> validationEvaluationLabels) |
TrainingConfig(IUpdater updater,
List<Regularization> regularization,
String dataSetFeatureMapping,
String dataSetLabelMapping)
Create a training configuration suitable for training a single input, single output network.
See also the TrainingConfig.Builder for creating a TrainingConfig |
| Modifier and Type | Interface and Description |
|---|---|
interface |
GradientUpdater<T extends IUpdater> |
| Modifier and Type | Class and Description |
|---|---|
class |
AdaBelief
AdaBelief
https://arxiv.org/pdf/2010.07468.pdf
|
class |
AdaDelta |
class |
AdaGrad |
class |
Adam
The Adam updater.
|
class |
AdaMax
The AdaMax updater, a variant of Adam.
|
class |
AMSGrad |
class |
Nadam |
class |
Nesterovs |
class |
NoOp |
class |
RmsProp |
class |
Sgd
SGD updater applies a learning rate only
|
| Modifier and Type | Method and Description |
|---|---|
IUpdater |
AdaMax.clone() |
IUpdater |
IUpdater.clone()
Clone the updater
|
Copyright © 2021. All rights reserved.