org.ea.javacnn.trainers

Class AdaGradTrainer

java.lang.Object

org.ea.javacnn.trainers.Trainer

org.ea.javacnn.trainers.AdaGradTrainer

public class AdaGradTrainer
extends Trainer

The adaptive gradient trainer will over time sum up the square of the gradient and use it to change the weights.

Constructor Summary

Constructors

Constructor and Description
AdaGradTrainer(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

AdaGradTrainer

public AdaGradTrainer(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