org.deeplearning4j.nn.conf.layers

Class GravesBidirectionalLSTM.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<GravesBidirectionalLSTM.Builder>

org.deeplearning4j.nn.conf.layers.GravesBidirectionalLSTM.Builder

Enclosing class:

GravesBidirectionalLSTM

public static class GravesBidirectionalLSTM.Builder
extends BaseRecurrentLayer.Builder<GravesBidirectionalLSTM.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
GravesBidirectionalLSTM	build()
GravesBidirectionalLSTM.Builder	forgetGateBiasInit(double biasInit) Set forget gate bias initalizations.
GravesBidirectionalLSTM.Builder	gateActivationFunction(org.nd4j.linalg.activations.Activation gateActivationFn) Activation function for the LSTM gates.
GravesBidirectionalLSTM.Builder	gateActivationFunction(org.nd4j.linalg.activations.IActivation gateActivationFn) Activation function for the LSTM gates.
GravesBidirectionalLSTM.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 GravesBidirectionalLSTM.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 GravesBidirectionalLSTM.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 GravesBidirectionalLSTM.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 GravesBidirectionalLSTM.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 GravesBidirectionalLSTM build()

Specified by:

build in class Layer.Builder<GravesBidirectionalLSTM.Builder>