weka.dl4j.layers

Class BatchNormalization

java.lang.Object

weka.dl4j.layers.Layer<T>

weka.dl4j.layers.FeedForwardLayer<org.deeplearning4j.nn.conf.layers.BatchNormalization>

weka.dl4j.layers.BatchNormalization

All Implemented Interfaces:

java.io.Serializable, weka.core.OptionHandler, ApiWrapper<org.deeplearning4j.nn.conf.layers.BatchNormalization>

public class BatchNormalization
extends FeedForwardLayer<org.deeplearning4j.nn.conf.layers.BatchNormalization>
implements weka.core.OptionHandler, java.io.Serializable

A version of DeepLearning4j's BatchNormalization layer that implements WEKA option handling.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
BatchNormalization() Constructor for setting some defaults.

Method Summary

Modifier and Type	Method and Description
double	getBeta()
double	getDecay()
double	getEps()
double	getGamma()
boolean	getIsMinibatch()
boolean	getLockGammaAndBeta()
long	getNOut() Deprecated.
java.lang.String[]	getOptions() Gets the current settings of the Classifier.
java.lang.String	globalInfo() Global info.
void	initializeBackend() Initialize the DL4J backend.
boolean	isLockGammaBeta()
java.util.Enumeration<weka.core.Option>	listOptions() Returns an enumeration describing the available options.
void	setBeta(double beta)
void	setDecay(double decay)
void	setEps(double eps)
void	setGamma(double gamma)
void	setLockGammaAndBeta(boolean lgb)
void	setLockGammaBeta(boolean lgb)
void	setMinibatch(boolean b)
void	setNOut(long nOut) Deprecated.
void	setOptions(java.lang.String[] options) Parses a given list of options.

Methods inherited from class weka.dl4j.layers.FeedForwardLayer

getActivationFunction, setActivationFunction

Methods inherited from class weka.dl4j.layers.Layer

create, getBackend, getLayerName, setBackend, setLayerName

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

BatchNormalization

public BatchNormalization()

Constructor for setting some defaults.

Method Detail

globalInfo

public java.lang.String globalInfo()

Global info.

Returns:

string describing this class.

getDecay

@OptionMetadata(displayName="decay parameter",
                description="The decay parameter (default = 0.9).",
                commandLineParamName="decay",
                commandLineParamSynopsis="-decay <double>",
                displayOrder=1)
public double getDecay()

setDecay

public void setDecay(double decay)

getEps

@OptionMetadata(displayName="eps parameter",
                description="The eps parameter (default = 1e-5).",
                commandLineParamName="eps",
                commandLineParamSynopsis="-eps <double>",
                displayOrder=2)
public double getEps()

setEps

public void setEps(double eps)

getGamma

@OptionMetadata(displayName="gamma parameter",
                description="The gamma parameter (default = 1).",
                commandLineParamName="gamma",
                commandLineParamSynopsis="-gamma <double>",
                displayOrder=3)
public double getGamma()

setGamma

public void setGamma(double gamma)

getBeta

@OptionMetadata(displayName="beta parameter",
                description="The beta parameter (default = 0).",
                commandLineParamName="beta",
                commandLineParamSynopsis="-beta <double>",
                displayOrder=4)
public double getBeta()

setBeta

public void setBeta(double beta)

getLockGammaAndBeta

@OptionMetadata(displayName="lock gamma and beta",
                description="Whether to lock gamma and beta.",
                commandLineParamName="beta",
                commandLineParamSynopsis="-lockGammaBeta",
                displayOrder=5)
public boolean getLockGammaAndBeta()

setLockGammaAndBeta

public void setLockGammaAndBeta(boolean lgb)

isLockGammaBeta

@ProgrammaticProperty
public boolean isLockGammaBeta()

setLockGammaBeta

public void setLockGammaBeta(boolean lgb)

getIsMinibatch

@OptionMetadata(displayName="isMinibatch",
                description="Whether minibatches are not not used.",
                commandLineParamName="isMinibatch",
                commandLineParamSynopsis="-isMinibatch",
                displayOrder=6)
public boolean getIsMinibatch()

setMinibatch

public void setMinibatch(boolean b)

getNOut

@ProgrammaticProperty
 @Deprecated
public long getNOut()

Deprecated.

Overrides:

getNOut in class FeedForwardLayer<org.deeplearning4j.nn.conf.layers.BatchNormalization>

setNOut

@ProgrammaticProperty
 @Deprecated
public void setNOut(long nOut)

Deprecated.

Overrides:

setNOut in class FeedForwardLayer<org.deeplearning4j.nn.conf.layers.BatchNormalization>

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 FeedForwardLayer<org.deeplearning4j.nn.conf.layers.BatchNormalization>

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 FeedForwardLayer<org.deeplearning4j.nn.conf.layers.BatchNormalization>

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 FeedForwardLayer<org.deeplearning4j.nn.conf.layers.BatchNormalization>

Parameters:

options - the list of options as an array of strings

Throws:

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

initializeBackend

public void initializeBackend()

Description copied from interface: ApiWrapper

Initialize the DL4J backend.

Specified by:

initializeBackend in interface ApiWrapper<org.deeplearning4j.nn.conf.layers.BatchNormalization>