public class RMSpropTrainer extends BackpropTrainer
currentEpoch, epochs, learnRate, net, shuffle, stepMode, testData, trainingData, trainingSignalGenerator| Constructor and Description |
|---|
RMSpropTrainer()
Creates an RMSprop trainer with a learn rate of 0.001, a batch size of 32.
|
| Modifier and Type | Method and Description |
|---|---|
protected double |
calcAdaptation(BrainPart part)
Calculates the RMSprop adaptation using a moving average of the squared gradient.
|
void |
fromXML(org.jdom2.Element element)
Reads attributes from XML.
|
double |
getEpsilon()
Returns the fuzz factor used to avoid numerical problems.
|
double |
getGamma()
Returns the value of the weight of the current squared gradient.
|
void |
setEpsilon(double epsilon)
Returns the fuzz factor used to avoid numerical problems.
|
void |
setGamma(double gamma)
Sets the value of the weight of the current squared gradient.
|
org.jdom2.Element |
toXML(org.jdom2.Element element)
Writes specific elements to XML tree.
|
endPattern, endTrain, reset, train, train, trainclone, endBatch, getBatchSize, getEpochs, getLearnRate, getNetwork, getRank, getTestData, getTrainingData, getTrainingSignalGenerator, getWinningNeuron, setBatchSize, setEpochs, setLearnRate, setNetwork, setShuffle, setStepMode, setTestData, setTrainingData, setTrainingSignalGenerator, test, test, toString, train, trainPattern, updateLinks, updateNeuronspublic RMSpropTrainer()
protected double calcAdaptation(BrainPart part)
calcAdaptation in class BackpropTrainerpart - the brain part to be adaptedpublic double getGamma()
public void setGamma(double gamma)
gamma - weight value (in unit interval)public double getEpsilon()
public void setEpsilon(double epsilon)
epsilon - the fuzz factor valuepublic org.jdom2.Element toXML(org.jdom2.Element element)