Class CnnLossLayer.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<CnnLossLayer.Builder>
-
- org.deeplearning4j.nn.conf.layers.CnnLossLayer.Builder
-
- Enclosing class:
- CnnLossLayer
public static class CnnLossLayer.Builder extends BaseOutputLayer.Builder<CnnLossLayer.Builder>
-
-
Field Summary
Fields Modifier and Type Field Description protected CNN2DFormat
format
-
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 CnnLossLayer
build()
CnnLossLayer.Builder
format(CNN2DFormat format)
CnnLossLayer.Builder
nIn(int nIn)
Number of inputs for the layer (usually the size of the last layer).CnnLossLayer.Builder
nOut(int nOut)
Number of outputs - used to set the layer size (number of units/nodes for the current layer).void
setNIn(long nIn)
void
setNOut(long nOut)
-
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
-
-
-
-
Field Detail
-
format
protected CNN2DFormat format
-
-
Constructor Detail
-
Builder
public Builder()
-
Builder
public Builder(LossFunctions.LossFunction lossFunction)
-
Builder
public Builder(ILossFunction lossFunction)
-
-
Method Detail
-
format
public CnnLossLayer.Builder format(CNN2DFormat format)
-
nIn
public CnnLossLayer.Builder nIn(int nIn)
Description copied from class:FeedForwardLayer.Builder
Number 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:
nIn
in classFeedForwardLayer.Builder<CnnLossLayer.Builder>
- Parameters:
nIn
- Number of inputs for the layer
-
nOut
public CnnLossLayer.Builder nOut(int nOut)
Description copied from class:FeedForwardLayer.Builder
Number 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:
nOut
in classFeedForwardLayer.Builder<CnnLossLayer.Builder>
- Parameters:
nOut
- Number of outputs / layer size
-
setNIn
public void setNIn(long nIn)
-
setNOut
public void setNOut(long nOut)
-
build
public CnnLossLayer build()
- Specified by:
build
in classLayer.Builder<CnnLossLayer.Builder>
-
-