org.deeplearning4j.arbiter.layers

Class BatchNormalizationSpace

java.lang.Object

org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace<L>

org.deeplearning4j.arbiter.layers.LayerSpace<L>

org.deeplearning4j.arbiter.layers.BaseLayerSpace<L>

org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace<org.deeplearning4j.nn.conf.layers.BatchNormalization>

org.deeplearning4j.arbiter.layers.BatchNormalizationSpace

All Implemented Interfaces:

ParameterSpace<org.deeplearning4j.nn.conf.layers.BatchNormalization>

public class BatchNormalizationSpace
extends FeedForwardLayerSpace<org.deeplearning4j.nn.conf.layers.BatchNormalization>

LayerSpace for batch normalization layers

Author:

Alex Black

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	BatchNormalizationSpace.Builder

Field Summary

Fields

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

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

constrainAll, constrainBias, constrainWeights, nIn, nOut

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

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

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

dropOut

Method Summary

Modifier and Type	Method and Description
org.deeplearning4j.nn.conf.layers.BatchNormalization	getValue(double[] parameterValues)
protected void	setLayerOptionsBuilder(org.deeplearning4j.nn.conf.layers.BatchNormalization.Builder builder, double[] values)
String	toString()
String	toString(String delim)

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

setLayerOptionsBuilder

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

isLeaf, numParameters, setIndices, setLayerOptionsBuilder

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

collectLeaves, setLayerOptionsBuilder

Methods inherited from class org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace

getNestedSpaces

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, 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

constrainBeta

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

constrainGamma

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

Method Detail

getValue

public org.deeplearning4j.nn.conf.layers.BatchNormalization getValue(double[] parameterValues)

setLayerOptionsBuilder

protected void setLayerOptionsBuilder(org.deeplearning4j.nn.conf.layers.BatchNormalization.Builder builder,
                                      double[] values)

toString

public String toString()

Overrides:

toString in class FeedForwardLayerSpace<org.deeplearning4j.nn.conf.layers.BatchNormalization>

toString

public String toString(String delim)

Overrides:

toString in class FeedForwardLayerSpace<org.deeplearning4j.nn.conf.layers.BatchNormalization>