org.deeplearning4j.nn.conf.layers

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

Direct Known Subclasses:

CenterLossOutputLayer.Builder, CnnLossLayer.Builder, LossLayer.Builder, OutputLayer.Builder, RnnLossLayer.Builder, RnnOutputLayer.Builder

Enclosing class:

BaseOutputLayer

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

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.lossfunctions.ILossFunction	lossFn

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, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, iupdater, l1, l1Bias, l2, l2Bias, weightInit, weightNoise

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

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

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
T	hasBias(boolean hasBias) If true (default): include bias parameters in the model.
T	lossFunction(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)
T	lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

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

nIn, nOut, units

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

activation, activation, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, updater, updater, weightInit, weightInit, weightNoise

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

build, constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name

Methods inherited from class java.lang.Object

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

Field Detail

lossFn

protected org.nd4j.linalg.lossfunctions.ILossFunction lossFn

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

lossFunction

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

hasBias

public T hasBias(boolean hasBias)

If true (default): include bias parameters in the model. False: no bias.

Parameters:

hasBias - If true: include bias parameters in this model

lossFunction

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