deepnetts.net.train.opt

Class SgdOptimizer

java.lang.Object

deepnetts.net.train.opt.SgdOptimizer

All Implemented Interfaces:

Optimizer, Serializable

public final class SgdOptimizer
extends Object
implements Optimizer, Serializable

Basic Stochastic Gradient Descent optimization algorithm, which iteratively change weights towards value which gives minimum error.

See Also:

Serialized Form

Field Summary

Fields inherited from interface deepnetts.net.train.opt.Optimizer

COL_IDX, DEPTH_IDX, EPS, FOURTH_DIM_IDX, ROW_IDX

Constructor Summary

Constructors

Constructor and Description
SgdOptimizer(AbstractLayer layer)

Method Summary

Modifier and Type	Method and Description
float	calculateDeltaBias(float gradient, int idx)
float	calculateDeltaWeight(float gradient, int... index)
void	setLearningRate(float learningRate)

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface deepnetts.net.train.opt.Optimizer

create

Constructor Detail

SgdOptimizer

public SgdOptimizer(AbstractLayer layer)

Method Detail

calculateDeltaWeight

public float calculateDeltaWeight(float gradient,
                                  int... index)

Specified by:

calculateDeltaWeight in interface Optimizer

calculateDeltaBias

public float calculateDeltaBias(float gradient,
                                int idx)

Specified by:

calculateDeltaBias in interface Optimizer

setLearningRate

public void setLearningRate(float learningRate)

Specified by:

setLearningRate in interface Optimizer