public class StochasticGradientDescent extends BaseOptimizer
computationGraphUpdater, conf, GRADIENT_KEY, iterationListeners, lineMaximizer, log, model, oldScore, PARAMS_KEY, score, SCORE_KEY, SEARCH_DIR, searchState, step, stepFunction, stepMax, terminationConditions, updater
Constructor and Description |
---|
StochasticGradientDescent(NeuralNetConfiguration conf,
StepFunction stepFunction,
java.util.Collection<IterationListener> iterationListeners,
java.util.Collection<TerminationCondition> terminationConditions,
Model model) |
StochasticGradientDescent(NeuralNetConfiguration conf,
StepFunction stepFunction,
java.util.Collection<IterationListener> iterationListeners,
Model model) |
Modifier and Type | Method and Description |
---|---|
boolean |
optimize()
Optimize call.
|
void |
postStep(org.nd4j.linalg.api.ndarray.INDArray gradient)
Post step to update searchDirection with new gradient and parameter information
|
void |
preProcessLine()
Pre preProcess to setup initial searchDirection approximation
|
batchSize, checkTerminalConditions, getComputationGraphUpdater, getConf, getDefaultStepFunctionForOptimizer, getIterationCount, getUpdater, gradientAndScore, incrementIterationCount, postFirstStep, score, setBatchSize, setListeners, setUpdater, setUpdaterComputationGraph, setupSearchState, updateGradientAccordingToParams
public StochasticGradientDescent(NeuralNetConfiguration conf, StepFunction stepFunction, java.util.Collection<IterationListener> iterationListeners, Model model)
public StochasticGradientDescent(NeuralNetConfiguration conf, StepFunction stepFunction, java.util.Collection<IterationListener> iterationListeners, java.util.Collection<TerminationCondition> terminationConditions, Model model)
public boolean optimize()
BaseOptimizer
optimize
in interface ConvexOptimizer
optimize
in class BaseOptimizer
public void preProcessLine()
BaseOptimizer
preProcessLine
in interface ConvexOptimizer
preProcessLine
in class BaseOptimizer
public void postStep(org.nd4j.linalg.api.ndarray.INDArray gradient)
BaseOptimizer
postStep
in interface ConvexOptimizer
postStep
in class BaseOptimizer