public class CenterLossOutputLayer extends BaseOutputLayer
Modifier and Type | Class and Description |
---|---|
static class |
CenterLossOutputLayer.Builder |
Modifier and Type | Field and Description |
---|---|
protected double |
alpha |
protected boolean |
gradientCheck |
protected double |
lambda |
lossFn
nIn, nOut
activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, layerName, learningRate, learningRateSchedule, momentum, momentumSchedule, rho, rmsDecay, updater, weightInit
Modifier | Constructor and Description |
---|---|
protected |
CenterLossOutputLayer(CenterLossOutputLayer.Builder builder) |
Modifier and Type | Method and Description |
---|---|
double |
getAlpha() |
boolean |
getGradientCheck() |
double |
getL1ByParam(java.lang.String paramName)
Get the L1 coefficient for the given parameter.
|
double |
getL2ByParam(java.lang.String paramName)
Get the L2 coefficient for the given parameter.
|
double |
getLambda() |
double |
getLearningRateByParam(java.lang.String paramName)
Get the (initial) learning rate coefficient for the given parameter.
|
Updater |
getUpdaterByParam(java.lang.String paramName)
Get the updater for the given parameter.
|
ParamInitializer |
initializer() |
Layer |
instantiate(NeuralNetConfiguration conf,
java.util.Collection<IterationListener> iterationListeners,
int layerIndex,
org.nd4j.linalg.api.ndarray.INDArray layerParamsView,
boolean initializeParams) |
getLossFunction
getOutputType, getPreProcessorForInputType, setNIn
clone, resetLayerDefaultConfig
protected double alpha
protected double lambda
protected boolean gradientCheck
protected CenterLossOutputLayer(CenterLossOutputLayer.Builder builder)
public Layer instantiate(NeuralNetConfiguration conf, java.util.Collection<IterationListener> iterationListeners, int layerIndex, org.nd4j.linalg.api.ndarray.INDArray layerParamsView, boolean initializeParams)
instantiate
in class Layer
public ParamInitializer initializer()
initializer
in class Layer
public Updater getUpdaterByParam(java.lang.String paramName)
Layer
getUpdaterByParam
in class Layer
paramName
- Parameter namepublic double getLearningRateByParam(java.lang.String paramName)
Layer
getLearningRateByParam
in class FeedForwardLayer
paramName
- Parameter namepublic double getL1ByParam(java.lang.String paramName)
Layer
getL1ByParam
in class FeedForwardLayer
paramName
- Parameter namepublic double getL2ByParam(java.lang.String paramName)
Layer
getL2ByParam
in class FeedForwardLayer
paramName
- Parameter namepublic double getAlpha()
public double getLambda()
public boolean getGradientCheck()