weka.dl4j.updater

Class RmsProp

java.lang.Object

weka.dl4j.updater.Updater<org.nd4j.linalg.learning.config.RmsProp>

weka.dl4j.updater.RmsProp

All Implemented Interfaces:

java.io.Serializable, weka.core.OptionHandler, ApiWrapper<org.nd4j.linalg.learning.config.RmsProp>

public class RmsProp
extends Updater<org.nd4j.linalg.learning.config.RmsProp>

A WEKA version of DeepLearning4j's RmsProp.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
RmsProp()

Method Summary

Modifier and Type	Method and Description
double	getEpsilon()
java.lang.String[]	getOptions() Gets the current settings of the Classifier.
double	getRmsDecay()
void	initializeBackend() Initialize the DL4J backend.
java.util.Enumeration<weka.core.Option>	listOptions() Returns an enumeration describing the available options.
void	setEpsilon(double epsilon)
void	setOptions(java.lang.String[] options) Parses a given list of options.
void	setRmsDecay(double rmsDecay)

Methods inherited from class weka.dl4j.updater.Updater

create, getBackend, getLearningRate, getLearningRateSchedule, hasLearningRate, setBackend, setLearningRate, setLearningRateSchedule

Methods inherited from class java.lang.Object

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

Methods inherited from interface weka.core.OptionHandler

makeCopy

Constructor Detail

RmsProp

public RmsProp()

Method Detail

getRmsDecay

@OptionMetadata(displayName="rmsDecay",
                description="The rms decay (default = 0.95).",
                commandLineParamName="rmsDecay",
                commandLineParamSynopsis="-rmsDecay <double>",
                displayOrder=1)
public double getRmsDecay()

setRmsDecay

public void setRmsDecay(double rmsDecay)

getEpsilon

@OptionMetadata(displayName="epsilon",
                description="The epsilon parameter (default = 1.0E-8).",
                commandLineParamName="epsilon",
                commandLineParamSynopsis="-epsilon <double>",
                displayOrder=2)
public double getEpsilon()

setEpsilon

public void setEpsilon(double epsilon)

initializeBackend

public void initializeBackend()

Description copied from interface: ApiWrapper

Initialize the DL4J backend.

listOptions

public java.util.Enumeration<weka.core.Option> listOptions()

Returns an enumeration describing the available options.

Specified by:

listOptions in interface weka.core.OptionHandler

Overrides:

listOptions in class Updater<org.nd4j.linalg.learning.config.RmsProp>

Returns:

an enumeration of all the available options.

getOptions

public java.lang.String[] getOptions()

Gets the current settings of the Classifier.

Specified by:

getOptions in interface weka.core.OptionHandler

Overrides:

getOptions in class Updater<org.nd4j.linalg.learning.config.RmsProp>

Returns:

an array of strings suitable for passing to setOptions

setOptions

public void setOptions(java.lang.String[] options)
                throws java.lang.Exception

Parses a given list of options.

Specified by:

setOptions in interface weka.core.OptionHandler

Overrides:

setOptions in class Updater<org.nd4j.linalg.learning.config.RmsProp>

Parameters:

options - the list of options as an array of strings

Throws:

java.lang.Exception - if an option is not supported