org.deeplearning4j.arbiter

Class BaseNetworkSpace<T>

java.lang.Object

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

org.deeplearning4j.arbiter.BaseNetworkSpace<T>

Type Parameters:

T - Type of network (MultiLayerNetwork or ComputationGraph)

All Implemented Interfaces:

ParameterSpace<T>

Direct Known Subclasses:

ComputationGraphSpace, MultiLayerSpace

public abstract class BaseNetworkSpace<T>
extends AbstractParameterSpace<T>

This is an abstract ParameterSpace for both MultiLayerNetworks (MultiLayerSpace) and ComputationGraph (ComputationGraphSpace)

Functionality here should match NeuralNetConfiguration.Builder

Author:

Alex Black

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static class	BaseNetworkSpace.Builder<T extends BaseNetworkSpace.Builder<T>>
static class	BaseNetworkSpace.LayerConf

Field Summary

Fields

Modifier and Type	Field and Description
protected ParameterSpace<org.nd4j.linalg.activations.IActivation>	activationFunction
protected ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>>	allParamConstraints
protected ParameterSpace<org.deeplearning4j.nn.conf.BackpropType>	backpropType
protected ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>>	biasConstraints
protected ParameterSpace<Double>	biasInit
protected ParameterSpace<org.nd4j.linalg.learning.config.IUpdater>	biasUpdater
protected ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode>	convolutionMode
protected ParameterSpace<org.deeplearning4j.nn.conf.distribution.Distribution>	dist
protected ParameterSpace<org.deeplearning4j.nn.conf.GradientNormalization>	gradientNormalization
protected ParameterSpace<Double>	gradientNormalizationThreshold
protected ParameterSpace<Double>	l1
protected ParameterSpace<Double>	l1Bias
protected ParameterSpace<Double>	l2
protected ParameterSpace<Double>	l2Bias
protected List<BaseNetworkSpace.LayerConf>	layerSpaces
protected ParameterSpace<Integer>	maxNumLineSearchIterations
protected ParameterSpace<Boolean>	miniBatch
protected ParameterSpace<Boolean>	minimize
protected int	numEpochs
protected ParameterSpace<org.deeplearning4j.nn.api.OptimizationAlgorithm>	optimizationAlgo
protected Long	seed
protected ParameterSpace<org.deeplearning4j.nn.conf.stepfunctions.StepFunction>	stepFunction
protected ParameterSpace<Integer>	tbpttBwdLength
protected ParameterSpace<Integer>	tbpttFwdLength
protected ParameterSpace<org.nd4j.linalg.learning.config.IUpdater>	updater
protected ParameterSpace<List<org.deeplearning4j.nn.api.layers.LayerConstraint>>	weightConstraints
protected ParameterSpace<org.deeplearning4j.nn.weights.WeightInit>	weightInit
protected ParameterSpace<org.deeplearning4j.nn.conf.weightnoise.IWeightNoise>	weightNoise

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseNetworkSpace()
protected	BaseNetworkSpace(BaseNetworkSpace.Builder builder)

Method Summary

Modifier and Type	Method and Description
List<ParameterSpace>	collectLeaves()
boolean	isLeaf()
protected org.deeplearning4j.nn.conf.NeuralNetConfiguration.Builder	randomGlobalConf(double[] values)
void	setIndices(int... indices)
String	toJson() Return a json configuration of this configuration space.
String	toString()
String	toYaml() Return a yaml configuration of this configuration space.

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

Methods inherited from interface org.deeplearning4j.arbiter.optimize.api.ParameterSpace

getValue, numParameters

Field Detail

seed

protected Long seed

optimizationAlgo

protected ParameterSpace<org.deeplearning4j.nn.api.OptimizationAlgorithm> optimizationAlgo

activationFunction

protected ParameterSpace<org.nd4j.linalg.activations.IActivation> activationFunction

biasInit

protected ParameterSpace<Double> biasInit

weightInit

protected ParameterSpace<org.deeplearning4j.nn.weights.WeightInit> weightInit

dist

protected ParameterSpace<org.deeplearning4j.nn.conf.distribution.Distribution> dist

maxNumLineSearchIterations

protected ParameterSpace<Integer> maxNumLineSearchIterations

miniBatch

protected ParameterSpace<Boolean> miniBatch

minimize

protected ParameterSpace<Boolean> minimize

stepFunction

protected ParameterSpace<org.deeplearning4j.nn.conf.stepfunctions.StepFunction> stepFunction

l1

protected ParameterSpace<Double> l1

l2

protected ParameterSpace<Double> l2

l1Bias

protected ParameterSpace<Double> l1Bias

l2Bias

protected ParameterSpace<Double> l2Bias

updater

protected ParameterSpace<org.nd4j.linalg.learning.config.IUpdater> updater

biasUpdater

protected ParameterSpace<org.nd4j.linalg.learning.config.IUpdater> biasUpdater

weightNoise

protected ParameterSpace<org.deeplearning4j.nn.conf.weightnoise.IWeightNoise> weightNoise

gradientNormalization

protected ParameterSpace<org.deeplearning4j.nn.conf.GradientNormalization> gradientNormalization

gradientNormalizationThreshold

protected ParameterSpace<Double> gradientNormalizationThreshold

convolutionMode

protected ParameterSpace<org.deeplearning4j.nn.conf.ConvolutionMode> convolutionMode

layerSpaces

protected List<BaseNetworkSpace.LayerConf> layerSpaces

backpropType

protected ParameterSpace<org.deeplearning4j.nn.conf.BackpropType> backpropType

tbpttFwdLength

protected ParameterSpace<Integer> tbpttFwdLength

tbpttBwdLength

protected ParameterSpace<Integer> tbpttBwdLength

allParamConstraints

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

weightConstraints

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

biasConstraints

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

numEpochs

protected int numEpochs

Constructor Detail

BaseNetworkSpace

protected BaseNetworkSpace(BaseNetworkSpace.Builder builder)

BaseNetworkSpace

protected BaseNetworkSpace()

Method Detail

randomGlobalConf

protected org.deeplearning4j.nn.conf.NeuralNetConfiguration.Builder randomGlobalConf(double[] values)

collectLeaves

public List<ParameterSpace> collectLeaves()

isLeaf

public boolean isLeaf()

setIndices

public void setIndices(int... indices)

toString

public String toString()

Overrides:

toString in class Object

toJson

public String toJson()

Return a json configuration of this configuration space.

Returns:

toYaml

public String toYaml()

Return a yaml configuration of this configuration space.

Returns: