org.deeplearning4j.arbiter.layers

Class BatchNormalizationSpace.Builder

java.lang.Object

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

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

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

org.deeplearning4j.arbiter.layers.BatchNormalizationSpace.Builder

Enclosing class:

BatchNormalizationSpace

public static class BatchNormalizationSpace.Builder
extends FeedForwardLayerSpace.Builder<BatchNormalizationSpace.Builder>

Field Summary

Fields

Modifier and Type	Field and Description
protected ParameterSpace<Double>	beta
protected ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>>	betaConstraints
protected ParameterSpace<Double>	decay
protected ParameterSpace<Double>	eps
protected ParameterSpace<Double>	gamma
protected ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>>	gammaConstraints
protected ParameterSpace<Boolean>	isMinibatch
protected ParameterSpace<Boolean>	lockGammaBeta

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
BatchNormalizationSpace.Builder	beta(double beta)
BatchNormalizationSpace.Builder	beta(ParameterSpace<Double> beta)
BatchNormalizationSpace	build()
BatchNormalizationSpace.Builder	constrainBeta(org.deeplearning4j.nn.api.layers.LayerConstraint... constraints)
BatchNormalizationSpace.Builder	constrainBeta(ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>> constraints)
BatchNormalizationSpace.Builder	constrainGamma(org.deeplearning4j.nn.api.layers.LayerConstraint... constraints)
BatchNormalizationSpace.Builder	constrainGamma(ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>> constraints)
BatchNormalizationSpace.Builder	decay(double decay)
BatchNormalizationSpace.Builder	decay(ParameterSpace<Double> decay)
BatchNormalizationSpace.Builder	eps(double eps)
BatchNormalizationSpace.Builder	eps(ParameterSpace<Double> eps)
BatchNormalizationSpace.Builder	gamma(double gamma)
BatchNormalizationSpace.Builder	gamma(ParameterSpace<Double> gamma)
BatchNormalizationSpace.Builder	lockGammaBeta(boolean lockGammaBeta)
BatchNormalizationSpace.Builder	lockGammaBeta(ParameterSpace<Boolean> lockGammaBeta)
BatchNormalizationSpace.Builder	minibatch(boolean minibatch)
BatchNormalizationSpace.Builder	minibatch(ParameterSpace<Boolean> minibatch)

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

dropOut, dropOut, iDropOut

Methods inherited from class java.lang.Object

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

Field Detail

decay

protected ParameterSpace<Double> decay

eps

protected ParameterSpace<Double> eps

isMinibatch

protected ParameterSpace<Boolean> isMinibatch

lockGammaBeta

protected ParameterSpace<Boolean> lockGammaBeta

gamma

protected ParameterSpace<Double> gamma

beta

protected ParameterSpace<Double> beta

betaConstraints

protected ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>> betaConstraints

gammaConstraints

protected ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>> gammaConstraints

Constructor Detail

Builder

public Builder()

Method Detail

minibatch

public BatchNormalizationSpace.Builder minibatch(boolean minibatch)

minibatch

public BatchNormalizationSpace.Builder minibatch(ParameterSpace<Boolean> minibatch)

gamma

public BatchNormalizationSpace.Builder gamma(double gamma)

gamma

public BatchNormalizationSpace.Builder gamma(ParameterSpace<Double> gamma)

beta

public BatchNormalizationSpace.Builder beta(double beta)

beta

public BatchNormalizationSpace.Builder beta(ParameterSpace<Double> beta)

eps

public BatchNormalizationSpace.Builder eps(double eps)

eps

public BatchNormalizationSpace.Builder eps(ParameterSpace<Double> eps)

decay

public BatchNormalizationSpace.Builder decay(double decay)

decay

public BatchNormalizationSpace.Builder decay(ParameterSpace<Double> decay)

lockGammaBeta

public BatchNormalizationSpace.Builder lockGammaBeta(boolean lockGammaBeta)

lockGammaBeta

public BatchNormalizationSpace.Builder lockGammaBeta(ParameterSpace<Boolean> lockGammaBeta)

constrainBeta

public BatchNormalizationSpace.Builder constrainBeta(org.deeplearning4j.nn.api.layers.LayerConstraint... constraints)

constrainBeta

public BatchNormalizationSpace.Builder constrainBeta(ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>> constraints)

constrainGamma

public BatchNormalizationSpace.Builder constrainGamma(org.deeplearning4j.nn.api.layers.LayerConstraint... constraints)

constrainGamma

public BatchNormalizationSpace.Builder constrainGamma(ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>> constraints)

build

public BatchNormalizationSpace build()

Specified by:

build in class LayerSpace.Builder<BatchNormalizationSpace.Builder>