ml.shifu.guagua.example.nn

Class Weight

java.lang.Object

ml.shifu.guagua.example.nn.Weight

public class Weight
extends Object

Weight is used to update NN weights according to propagation option. Which is also copied from Encog.

We'd like to reuse code from Encog but unfortunately the methods are private:(.

Constructor Summary

Constructors

Constructor and Description
Weight(int numWeight, double numTrainSize, double rate, String algorithm)

Method Summary

Modifier and Type	Method and Description
double[]	calculateWeights(double[] weights, double[] gradients)

Methods inherited from class java.lang.Object

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

Constructor Detail

Weight

public Weight(int numWeight,
              double numTrainSize,
              double rate,
              String algorithm)

Method Detail

calculateWeights

public double[] calculateWeights(double[] weights,
                                 double[] gradients)