org.deeplearning4j.nn.conf.layers

Class GravesLSTM.Builder

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<T>

org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.BaseRecurrentLayer.Builder<GravesLSTM.Builder>

org.deeplearning4j.nn.conf.layers.GravesLSTM.Builder

Enclosing class:

GravesLSTM

public static class GravesLSTM.Builder
extends BaseRecurrentLayer.Builder<GravesLSTM.Builder>

Field Summary

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

nIn, nOut

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

activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, layerName, learningRate, learningRatePolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, weightInit

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
GravesLSTM	build()
GravesLSTM.Builder	forgetGateBiasInit(double biasInit) Set forget gate bias initalizations.
GravesLSTM.Builder	gateActivationFunction(org.nd4j.linalg.activations.Activation gateActivationFn) Activation function for the LSTM gates.
GravesLSTM.Builder	gateActivationFunction(org.nd4j.linalg.activations.IActivation gateActivationFn) Activation function for the LSTM gates.
GravesLSTM.Builder	gateActivationFunction(java.lang.String gateActivationFn) Activation function for the LSTM gates.

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

nIn, nOut

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

activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, name, rho, rmsDecay, updater, weightInit

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

forgetGateBiasInit

public GravesLSTM.Builder forgetGateBiasInit(double biasInit)

Set forget gate bias initalizations. Values in range 1-5 can potentially help with learning or longer-term dependencies.

gateActivationFunction

public GravesLSTM.Builder gateActivationFunction(java.lang.String gateActivationFn)

Activation function for the LSTM gates. Note: This should be bounded to range 0-1: sigmoid or hard sigmoid, for example

Parameters:

gateActivationFn - Activation function for the LSTM gates

gateActivationFunction

public GravesLSTM.Builder gateActivationFunction(org.nd4j.linalg.activations.Activation gateActivationFn)

Activation function for the LSTM gates. Note: This should be bounded to range 0-1: sigmoid or hard sigmoid, for example

Parameters:

gateActivationFn - Activation function for the LSTM gates

gateActivationFunction

public GravesLSTM.Builder gateActivationFunction(org.nd4j.linalg.activations.IActivation gateActivationFn)

Activation function for the LSTM gates. Note: This should be bounded to range 0-1: sigmoid or hard sigmoid, for example

Parameters:

gateActivationFn - Activation function for the LSTM gates

build

public GravesLSTM build()

Specified by:

build in class Layer.Builder<GravesLSTM.Builder>