Class RnnOutputLayer.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<RnnOutputLayer.Builder>
-
- org.deeplearning4j.nn.conf.layers.RnnOutputLayer.Builder
-
- Enclosing class:
- RnnOutputLayer
public static class RnnOutputLayer.Builder extends BaseOutputLayer.Builder<RnnOutputLayer.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 RnnOutputLayer
build()
RnnOutputLayer.Builder
dataFormat(RNNFormat rnnDataFormat)
-
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)
- Parameters:
lossFunction
- Loss function for the output layer
-
Builder
public Builder(ILossFunction lossFunction)
- Parameters:
lossFunction
- Loss function for the output layer
-
-
Method Detail
-
build
public RnnOutputLayer build()
- Specified by:
build
in classLayer.Builder<RnnOutputLayer.Builder>
-
dataFormat
public RnnOutputLayer.Builder dataFormat(RNNFormat rnnDataFormat)
- Parameters:
rnnDataFormat
- Data format expected by the layer. NCW = [miniBatchSize, size, timeSeriesLength], NWC = [miniBatchSize, timeSeriesLength, size]. Defaults to NCW.
-
-