org.deeplearning4j.nn.conf.layers

Class BasePretrainNetwork.Builder<T extends BasePretrainNetwork.Builder<T>>

java.lang.Object

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

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

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

Direct Known Subclasses:

AutoEncoder.Builder, RBM.Builder

Enclosing class:

BasePretrainNetwork

public abstract static class BasePretrainNetwork.Builder<T extends BasePretrainNetwork.Builder<T>>
extends FeedForwardLayer.Builder<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected String	customLossFunction
protected org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction	lossFunction

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

nIn, nOut

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

activationFunction, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l2, 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
T	customLossFunction(String customLossFunction)
T	lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

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, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, build, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l2, 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

Field Detail

lossFunction

protected org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction

customLossFunction

protected String customLossFunction

Constructor Detail

Builder

public Builder()

Method Detail

lossFunction

public T lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

customLossFunction

public T customLossFunction(String customLossFunction)