Class LossLayer.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<LossLayer.Builder>
-
- org.deeplearning4j.nn.conf.layers.LossLayer.Builder
-
- Enclosing class:
- LossLayer
public static class LossLayer.Builder extends BaseOutputLayer.Builder<LossLayer.Builder>
-
-
Field Summary
-
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 LossLayerbuild()LossLayer.BuildernIn(int nIn)Number of inputs for the layer (usually the size of the last layer).LossLayer.BuildernOut(int nOut)Number of outputs - used to set the layer size (number of units/nodes for the current layer).-
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, 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)
- Parameters:
lossFunction- Loss function for the loss layer
-
Builder
public Builder(ILossFunction lossFunction)
- Parameters:
lossFunction- Loss function for the loss layer
-
-
Method Detail
-
nIn
public LossLayer.Builder nIn(int nIn)
Description copied from class:FeedForwardLayer.BuilderNumber of inputs for the layer (usually the size of the last layer).
Note that for Convolutional layers, this is the input channels, otherwise is the previous layer size.- Overrides:
nInin classFeedForwardLayer.Builder<LossLayer.Builder>- Parameters:
nIn- Number of inputs for the layer
-
nOut
public LossLayer.Builder nOut(int nOut)
Description copied from class:FeedForwardLayer.BuilderNumber of outputs - used to set the layer size (number of units/nodes for the current layer). Note that this is equivalent toFeedForwardLayer.Builder.units(int)- Overrides:
nOutin classFeedForwardLayer.Builder<LossLayer.Builder>- Parameters:
nOut- Number of outputs / layer size
-
build
public LossLayer build()
- Specified by:
buildin classLayer.Builder<LossLayer.Builder>
-
-