org.deeplearning4j.nn.conf.layers

Class AbstractLSTM.Builder<T extends AbstractLSTM.Builder<T>>

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.BaseRecurrentLayer.Builder<T>

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

Direct Known Subclasses:

GravesLSTM.Builder, LSTM.Builder

Enclosing class:

AbstractLSTM

public abstract static class AbstractLSTM.Builder<T extends AbstractLSTM.Builder<T>>
extends BaseRecurrentLayer.Builder<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected double	forgetGateBiasInit
protected org.nd4j.linalg.activations.IActivation	gateActivationFn

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, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, epsilon, gradientNormalization, gradientNormalizationThreshold, iupdater, l1, l1Bias, l2, l2Bias, learningRate, learningRatePolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, weightInit

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

dropOut, layerName

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
T	forgetGateBiasInit(double biasInit) Set forget gate bias initalizations.
T	gateActivationFunction(org.nd4j.linalg.activations.Activation gateActivationFn) Activation function for the LSTM gates.
T	gateActivationFunction(org.nd4j.linalg.activations.IActivation gateActivationFn) Activation function for the LSTM gates.
T	gateActivationFunction(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.BaseLayer.Builder

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

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

build, dropOut, name

Methods inherited from class java.lang.Object

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

Field Detail

forgetGateBiasInit

protected double forgetGateBiasInit

gateActivationFn

protected org.nd4j.linalg.activations.IActivation gateActivationFn

Constructor Detail

Builder

public Builder()

Method Detail

forgetGateBiasInit

public T forgetGateBiasInit(double biasInit)

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

gateActivationFunction

public T gateActivationFunction(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 T 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 T 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