org.ea.javacnn.trainers

Class SGDTrainer

java.lang.Object

org.ea.javacnn.trainers.Trainer

org.ea.javacnn.trainers.SGDTrainer

public class SGDTrainer
extends Trainer

Stochastic gradient descent (often shortened in SGD), also known as incremental gradient descent, is a stochastic approximation of the gradient descent optimization method for minimizing an objective function that is written as a sum of differentiable functions. In other words, SGD tries to find minimums or maximums by iteration.

Constructor Summary

Constructors

Constructor and Description
SGDTrainer(JavaCNN net, int batch_size, float l2_decay)

Method Summary

Modifier and Type	Method and Description
void	update(int i, int j, double gij, double[] p)

Methods inherited from class org.ea.javacnn.trainers.Trainer

initTrainData, train

Methods inherited from class java.lang.Object

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

Constructor Detail

SGDTrainer

public SGDTrainer(JavaCNN net,
                  int batch_size,
                  float l2_decay)

Method Detail

update

public void update(int i,
                   int j,
                   double gij,
                   double[] p)

Specified by:

update in class Trainer