Class CenterLossOutputLayer.Builder
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.BaseLayer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.BaseOutputLayer.Builder<CenterLossOutputLayer.Builder>
-
- org.deeplearning4j.nn.conf.layers.CenterLossOutputLayer.Builder
-
- Enclosing class:
- CenterLossOutputLayer
public static class CenterLossOutputLayer.Builder extends BaseOutputLayer.Builder<CenterLossOutputLayer.Builder>
-
-
Field Summary
Fields Modifier and Type Field Description protected double
alpha
protected boolean
gradientCheck
protected double
lambda
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseOutputLayer.Builder
lossFn
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder
nIn, nOut
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iupdater, regularization, regularizationBias, weightInitFn, weightNoise
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
-
-
Constructor Summary
Constructors Constructor Description Builder()
Builder(ILossFunction lossFunction)
Builder(LossFunctions.LossFunction lossFunction)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description CenterLossOutputLayer.Builder
alpha(double alpha)
CenterLossOutputLayer
build()
CenterLossOutputLayer.Builder
gradientCheck(boolean isGradientCheck)
CenterLossOutputLayer.Builder
lambda(double lambda)
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseOutputLayer.Builder
hasBias, lossFunction, lossFunction
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder
nIn, nIn, nOut, nOut, units
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder
activation, activation, biasInit, biasUpdater, dist, gainInit, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias, weightInit, weightInit, weightInit, weightNoise
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
-
-
-
-
Constructor Detail
-
Builder
public Builder()
-
Builder
public Builder(LossFunctions.LossFunction lossFunction)
-
Builder
public Builder(ILossFunction lossFunction)
-
-
Method Detail
-
alpha
public CenterLossOutputLayer.Builder alpha(double alpha)
-
lambda
public CenterLossOutputLayer.Builder lambda(double lambda)
-
gradientCheck
public CenterLossOutputLayer.Builder gradientCheck(boolean isGradientCheck)
-
build
public CenterLossOutputLayer build()
- Specified by:
build
in classLayer.Builder<CenterLossOutputLayer.Builder>
-
-