org.deeplearning4j.nn.conf.layers

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 and Description
Builder()
Builder(ILossFunction lossFunction)
Builder(LossFunctions.LossFunction lossFunction)

Method Summary

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

Methods inherited from class java.lang.Object

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

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 class Layer.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.