org.deeplearning4j.arbiter

Class MultiLayerSpace.Builder

java.lang.Object

org.deeplearning4j.arbiter.BaseNetworkSpace.Builder<MultiLayerSpace.Builder>

org.deeplearning4j.arbiter.MultiLayerSpace.Builder

Enclosing class:

MultiLayerSpace

public static class MultiLayerSpace.Builder
extends BaseNetworkSpace.Builder<MultiLayerSpace.Builder>

Field Summary

Fields

Modifier and Type	Field and Description
protected org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.multilayer.MultiLayerNetwork>	earlyStoppingConfiguration
protected org.deeplearning4j.nn.conf.WorkspaceMode	inferenceWorkspaceMode
protected ParameterSpace<Map<Integer,org.deeplearning4j.nn.conf.InputPreProcessor>>	inputPreProcessors
protected ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType>	inputType
protected List<BaseNetworkSpace.LayerConf>	layerSpaces
protected org.deeplearning4j.nn.conf.WorkspaceMode	trainingWorkspaceMode

Fields inherited from class org.deeplearning4j.arbiter.BaseNetworkSpace.Builder

validateOutputLayerConfig

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
MultiLayerSpace.Builder	addLayer(LayerSpace<?> layerSpace)
MultiLayerSpace.Builder	addLayer(LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace, ParameterSpace<Integer> numLayersDistribution)
MultiLayerSpace.Builder	addLayer(LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace, ParameterSpace<Integer> numLayersDistribution, boolean duplicateConfig) Deprecated.
MultiLayerSpace	build()
MultiLayerSpace.Builder	earlyStoppingConfiguration(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.multilayer.MultiLayerNetwork> earlyStoppingConfiguration) Early stopping configuration (optional).
MultiLayerSpace.Builder	inferenceWorkspaceMode(org.deeplearning4j.nn.conf.WorkspaceMode workspaceMode)
MultiLayerSpace.Builder	layer(org.deeplearning4j.nn.conf.layers.Layer layer)
MultiLayerSpace.Builder	layer(LayerSpace<?> layerSpace)
MultiLayerSpace.Builder	layer(LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace, ParameterSpace<Integer> numLayersDistribution)
MultiLayerSpace.Builder	setInputPreProcessors(Map<Integer,org.deeplearning4j.nn.conf.InputPreProcessor> inputPreProcessors)
MultiLayerSpace.Builder	setInputPreProcessors(ParameterSpace<Map<Integer,org.deeplearning4j.nn.conf.InputPreProcessor>> inputPreProcessors)
MultiLayerSpace.Builder	setInputType(org.deeplearning4j.nn.conf.inputs.InputType inputType)
MultiLayerSpace.Builder	setInputType(ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType> inputType)
MultiLayerSpace.Builder	trainingWorkspaceMode(org.deeplearning4j.nn.conf.WorkspaceMode workspaceMode)

Methods inherited from class org.deeplearning4j.arbiter.BaseNetworkSpace.Builder

activation, activation, activationFn, backpropType, backpropType, biasInit, biasInit, biasUpdater, biasUpdater, constrainAllParams, constrainAllParams, constrainBias, constrainBias, constrainWeights, constrainWeights, convolutionMode, convolutionMode, dist, dist, dropOut, dropOut, gradientNormalization, gradientNormalization, gradientNormalizationThreshold, gradientNormalizationThreshold, idropOut, idropOut, l1, l1, l1Bias, l1Bias, l2, l2, l2Bias, l2Bias, maxNumLineSearchIterations, maxNumLineSearchIterations, miniBatch, miniBatch, minimize, minimize, numEpochs, optimizationAlgo, optimizationAlgo, seed, stepFunction, stepFunction, tbpttBwdLength, tbpttBwdLength, tbpttFwdLength, tbpttFwdLength, updater, updater, validateOutputLayerConfig, weightInit, weightInit, weightNoise, weightNoise

Methods inherited from class java.lang.Object

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

Field Detail

layerSpaces

protected List<BaseNetworkSpace.LayerConf> layerSpaces

inputType

protected ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType> inputType

inputPreProcessors

protected ParameterSpace<Map<Integer,org.deeplearning4j.nn.conf.InputPreProcessor>> inputPreProcessors

trainingWorkspaceMode

protected org.deeplearning4j.nn.conf.WorkspaceMode trainingWorkspaceMode

inferenceWorkspaceMode

protected org.deeplearning4j.nn.conf.WorkspaceMode inferenceWorkspaceMode

earlyStoppingConfiguration

protected org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.multilayer.MultiLayerNetwork> earlyStoppingConfiguration

Constructor Detail

Builder

public Builder()

Method Detail

setInputType

public MultiLayerSpace.Builder setInputType(org.deeplearning4j.nn.conf.inputs.InputType inputType)

setInputType

public MultiLayerSpace.Builder setInputType(ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType> inputType)

layer

public MultiLayerSpace.Builder layer(org.deeplearning4j.nn.conf.layers.Layer layer)

layer

public MultiLayerSpace.Builder layer(LayerSpace<?> layerSpace)

layer

public MultiLayerSpace.Builder layer(LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace,
                                     ParameterSpace<Integer> numLayersDistribution)

addLayer

public MultiLayerSpace.Builder addLayer(LayerSpace<?> layerSpace)

addLayer

@Deprecated
public MultiLayerSpace.Builder addLayer(LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace,
                                                    ParameterSpace<Integer> numLayersDistribution,
                                                    boolean duplicateConfig)

Deprecated.

duplicateConfig not supported. Will always be true

Parameters:

layerSpace -

numLayersDistribution -

duplicateConfig -

Returns:

addLayer

public MultiLayerSpace.Builder addLayer(LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace,
                                        ParameterSpace<Integer> numLayersDistribution)

Parameters:

layerSpace -

numLayersDistribution - Distribution for number of layers to generate

earlyStoppingConfiguration

public MultiLayerSpace.Builder earlyStoppingConfiguration(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.multilayer.MultiLayerNetwork> earlyStoppingConfiguration)

Early stopping configuration (optional). Note if both EarlyStoppingConfiguration and number of epochs is present, early stopping will be used in preference.

setInputPreProcessors

public MultiLayerSpace.Builder setInputPreProcessors(Map<Integer,org.deeplearning4j.nn.conf.InputPreProcessor> inputPreProcessors)

Parameters:

inputPreProcessors - Input preprocessors to set for the model

setInputPreProcessors

public MultiLayerSpace.Builder setInputPreProcessors(ParameterSpace<Map<Integer,org.deeplearning4j.nn.conf.InputPreProcessor>> inputPreProcessors)

Parameters:

inputPreProcessors - Input preprocessors to set for the model

trainingWorkspaceMode

public MultiLayerSpace.Builder trainingWorkspaceMode(org.deeplearning4j.nn.conf.WorkspaceMode workspaceMode)

inferenceWorkspaceMode

public MultiLayerSpace.Builder inferenceWorkspaceMode(org.deeplearning4j.nn.conf.WorkspaceMode workspaceMode)

build

public MultiLayerSpace build()

Specified by:

build in class BaseNetworkSpace.Builder<MultiLayerSpace.Builder>