org.deeplearning4j.nn.conf.layers

Class OutputLayer.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.BaseOutputLayer.Builder<OutputLayer.Builder>

org.deeplearning4j.nn.conf.layers.OutputLayer.Builder

Enclosing class:

OutputLayer

public static class OutputLayer.Builder
extends BaseOutputLayer.Builder<OutputLayer.Builder>

Field Summary

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

customLossFunction, lossFn

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(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)
Builder(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

Method Summary

Modifier and Type	Method and Description
OutputLayer	build()

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

customLossFunction, 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, 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

Constructor Detail

Builder

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

Builder

public Builder(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)

Method Detail

build

public OutputLayer build()

Specified by:

build in class Layer.Builder<OutputLayer.Builder>