public class GradientAdjustment extends Object
Modifier and Type | Method and Description |
---|---|
static void |
updateGradientAccordingToParams(int iteration,
int batchSize,
NeuralNetConfiguration conf,
org.nd4j.linalg.api.ndarray.INDArray params,
org.nd4j.linalg.api.ndarray.INDArray gradient,
org.nd4j.linalg.learning.GradientUpdater adaGrad,
org.nd4j.linalg.api.ndarray.INDArray lastStep,
String paramType)
Deprecated.
|
@Deprecated public static void updateGradientAccordingToParams(int iteration, int batchSize, NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.INDArray params, org.nd4j.linalg.api.ndarray.INDArray gradient, org.nd4j.linalg.learning.GradientUpdater adaGrad, org.nd4j.linalg.api.ndarray.INDArray lastStep, String paramType)
iteration
- the iterationbatchSize
- the batch size of the inputconf
- the configurationparams
- the model paramsgradient
- the gradients for the variablesadaGrad
- the adagrad map (per variable adagrad entries(Copyright © 2015. All Rights Reserved.