weka.dl4j.layers

Class CenterLossOutputLayer

java.lang.Object

weka.dl4j.layers.Layer<T>

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

weka.dl4j.layers.CenterLossOutputLayer

All Implemented Interfaces:

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

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

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

See Also:

Serialized Form

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
double	getAlpha()
boolean	getGradientCheck()
double	getLambda()
LossFunction<? extends org.nd4j.linalg.lossfunctions.ILossFunction>	getLossFn()
java.lang.String[]	getOptions() Gets the current settings of the Classifier.
java.lang.String	globalInfo() Global info.
void	initializeBackend() Initialize the DL4J backend.
java.util.Enumeration<weka.core.Option>	listOptions() Returns an enumeration describing the available options.
void	setAlpha(double alpha)
void	setGradientCheck(boolean gradientCheck)
void	setLambda(double lambda)
void	setLossFn(LossFunction<? extends org.nd4j.linalg.lossfunctions.ILossFunction> lossFn)
void	setOptions(java.lang.String[] options) Parses a given list of options.

Methods inherited from class weka.dl4j.layers.FeedForwardLayer

getActivationFunction, getNOut, setActivationFunction, setNOut

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

CenterLossOutputLayer

public CenterLossOutputLayer()

Constructor for setting some defaults.

Method Detail

getLossFn

@OptionMetadata(displayName="loss function",
                description="The loss function to use (default = LossMCXENT).",
                commandLineParamName="lossFn",
                commandLineParamSynopsis="-lossFn <specification>",
                displayOrder=1)
public LossFunction<? extends org.nd4j.linalg.lossfunctions.ILossFunction> getLossFn()

setLossFn

public void setLossFn(LossFunction<? extends org.nd4j.linalg.lossfunctions.ILossFunction> lossFn)

getGradientCheck

@OptionMetadata(displayName="gradient check",
                description="Enable/disable gradient checks (default = false).",
                commandLineParamName="gradientCheck",
                commandLineParamSynopsis="-gradientCheck <boolean>",
                displayOrder=1)
public boolean getGradientCheck()

setGradientCheck

public void setGradientCheck(boolean gradientCheck)

getAlpha

@OptionMetadata(displayName="alpha",
                description="The alpha value (default = 0.05).",
                commandLineParamName="eps",
                commandLineParamSynopsis="-eps <double>",
                displayOrder=2)
public double getAlpha()

setAlpha

public void setAlpha(double alpha)

getLambda

@OptionMetadata(displayName="lambda",
                description="The lambda value (default = 2e-4).",
                commandLineParamName="lambda",
                commandLineParamSynopsis="-lambda <double>",
                displayOrder=2)
public double getLambda()

setLambda

public void setLambda(double lambda)

globalInfo

public java.lang.String globalInfo()

Global info.

Returns:

string describing this class.

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.CenterLossOutputLayer>

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.CenterLossOutputLayer>

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.CenterLossOutputLayer>

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.CenterLossOutputLayer>