org.deeplearning4j.nn.conf.layers

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

org.deeplearning4j.nn.conf.layers.LossLayer.Builder

Enclosing class:

LossLayer

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

Field Summary

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

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

Method Summary

Modifier and Type	Method and Description
LossLayer	build()
LossLayer.Builder	nIn(int nIn)
LossLayer.Builder	nOut(int nOut)

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

lossFunction, lossFunction

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()

Builder

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

Builder

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

Method Detail

nIn

public LossLayer.Builder nIn(int nIn)

Overrides:

nIn in class FeedForwardLayer.Builder<LossLayer.Builder>

nOut

public LossLayer.Builder nOut(int nOut)

Overrides:

nOut in class FeedForwardLayer.Builder<LossLayer.Builder>

build

public LossLayer build()

Specified by:

build in class Layer.Builder<LossLayer.Builder>