org.deeplearning4j.nn.conf.layers

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 and 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, dist, gradientNormalization, gradientNormalizationThreshold, iupdater, l1, l1Bias, l2, l2Bias, weightInit, weightNoise

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)
Builder(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

Method Summary

Modifier and Type	Method and 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, nOut, units

Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder

activation, activation, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, updater, updater, weightInit, weightInit, weightNoise

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

alpha

protected double alpha

lambda

protected double lambda

gradientCheck

protected boolean gradientCheck

Constructor Detail

Builder

public Builder(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

Builder

public Builder(org.nd4j.linalg.lossfunctions.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 class Layer.Builder<CenterLossOutputLayer.Builder>