org.deeplearning4j.arbiter.layers

Class BaseOutputLayerSpace.Builder<T>

java.lang.Object

org.deeplearning4j.arbiter.layers.LayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.BaseOutputLayerSpace.Builder<T>

Direct Known Subclasses:

CenterLossOutputLayerSpace.Builder, OCNNLayerSpace.Builder, OutputLayerSpace.Builder, RnnOutputLayerSpace.Builder

Enclosing class:

BaseOutputLayerSpace<L extends org.deeplearning4j.nn.conf.layers.BaseOutputLayer>

public abstract static class BaseOutputLayerSpace.Builder<T>
extends FeedForwardLayerSpace.Builder<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected ParameterSpace<Boolean>	hasBias
protected ParameterSpace<org.nd4j.linalg.lossfunctions.ILossFunction>	lossFunction

Fields inherited from class org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder

constrainAll, constrainBias, constrainWeights, nIn, nOut

Fields inherited from class org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder

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

Fields inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

dropOut

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
T	hasBias(boolean hasBias)
T	hasBias(ParameterSpace<Boolean> hasBias)
T	iLossFunction(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)
T	iLossFunction(ParameterSpace<org.nd4j.linalg.lossfunctions.ILossFunction> lossFunction)
T	lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
T	lossFunction(ParameterSpace<org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction> lossFunction)

Methods inherited from class org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder

constrainAllParams, constrainAllParams, constrainBias, constrainBias, constrainWeights, constrainWeights, nIn, nIn, nOut, nOut

Methods inherited from class org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder

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

Methods inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

build, dropOut, dropOut, iDropOut

Methods inherited from class java.lang.Object

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

Field Detail

lossFunction

protected ParameterSpace<org.nd4j.linalg.lossfunctions.ILossFunction> lossFunction

hasBias

protected ParameterSpace<Boolean> hasBias

Constructor Detail

Builder

public Builder()

Method Detail

lossFunction

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

lossFunction

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

iLossFunction

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

iLossFunction

public T iLossFunction(ParameterSpace<org.nd4j.linalg.lossfunctions.ILossFunction> lossFunction)

hasBias

public T hasBias(boolean hasBias)

hasBias

public T hasBias(ParameterSpace<Boolean> hasBias)