public static class CenterLossOutputLayer.Builder extends BaseOutputLayer.Builder<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, learningRatePolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, weightInit
Constructor and Description |
---|
Builder(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction) |
Builder(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction) |
Modifier and Type | Method and Description |
---|---|
CenterLossOutputLayer.Builder |
alpha(double alpha) |
CenterLossOutputLayer |
build() |
CenterLossOutputLayer.Builder |
gradientCheck(boolean isGradientCheck) |
CenterLossOutputLayer.Builder |
lambda(double lambda) |
lossFunction, lossFunction
nIn, nOut
activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, name, rho, rmsDecay, updater, weightInit
protected double alpha
protected double lambda
protected boolean gradientCheck
public Builder(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
public Builder(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)
public CenterLossOutputLayer.Builder alpha(double alpha)
public CenterLossOutputLayer.Builder lambda(double lambda)
public CenterLossOutputLayer.Builder gradientCheck(boolean isGradientCheck)
public CenterLossOutputLayer build()
build
in class Layer.Builder<CenterLossOutputLayer.Builder>