org.deeplearning4j.nn.layers.custom.testclasses

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

org.deeplearning4j.nn.layers.custom.testclasses.CustomOutputLayer.Builder

Enclosing class:

CustomOutputLayer

public static class CustomOutputLayer.Builder
extends BaseOutputLayer.Builder<CustomOutputLayer.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

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

Method Summary

Modifier and Type	Method and Description
CustomOutputLayer	build()

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

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

Method Detail

build

public CustomOutputLayer build()

Specified by:

build in class Layer.Builder<CustomOutputLayer.Builder>