org.deeplearning4j.arbiter

Class ComputationGraphSpace.Builder

java.lang.Object

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

org.deeplearning4j.arbiter.ComputationGraphSpace.Builder

Enclosing class:

ComputationGraphSpace

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

Field Summary

Fields

Modifier and Type	Field and Description
protected org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.graph.ComputationGraph>	earlyStoppingConfiguration
protected org.deeplearning4j.nn.conf.WorkspaceMode	inferenceWorkspaceMode
protected ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType[]>	inputTypes
protected List<BaseNetworkSpace.LayerConf>	layerList
protected String[]	networkInputs
protected String[]	networkOutputs
protected org.deeplearning4j.nn.conf.WorkspaceMode	trainingWorkspaceMode
protected List<ComputationGraphSpace.VertexConf>	vertexList

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
ComputationGraphSpace.Builder	addInputs(String... networkInputs)
ComputationGraphSpace.Builder	addLayer(String layerName, LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace, org.deeplearning4j.nn.conf.InputPreProcessor preProcessor, String... layerInputs)
ComputationGraphSpace.Builder	addLayer(String layerName, LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace, String... layerInputs)
ComputationGraphSpace.Builder	addVertex(String vertexName, org.deeplearning4j.nn.conf.graph.GraphVertex vertex, String... vertexInputs)
ComputationGraphSpace	build()
ComputationGraphSpace.Builder	earlyStoppingConfiguration(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.graph.ComputationGraph> earlyStoppingConfiguration) Early stopping configuration (optional).
ComputationGraphSpace.Builder	inferenceWorkspaceMode(org.deeplearning4j.nn.conf.WorkspaceMode workspaceMode)
ComputationGraphSpace.Builder	layer(String layerName, LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace, org.deeplearning4j.nn.conf.InputPreProcessor preProcessor, String... layerInputs)
ComputationGraphSpace.Builder	layer(String layerName, LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace, String... layerInputs)
ComputationGraphSpace.Builder	layer(String layerName, org.deeplearning4j.nn.conf.layers.Layer layer, String... layerInputs)
ComputationGraphSpace.Builder	setInputTypes(org.deeplearning4j.nn.conf.inputs.InputType... inputTypes)
ComputationGraphSpace.Builder	setInputTypes(ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType[]> inputTypes)
ComputationGraphSpace.Builder	setOutputs(String... networkOutputs)
ComputationGraphSpace.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

layerList

protected List<BaseNetworkSpace.LayerConf> layerList

vertexList

protected List<ComputationGraphSpace.VertexConf> vertexList

earlyStoppingConfiguration

protected org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.graph.ComputationGraph> earlyStoppingConfiguration

networkInputs

protected String[] networkInputs

networkOutputs

protected String[] networkOutputs

inputTypes

protected ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType[]> inputTypes

trainingWorkspaceMode

protected org.deeplearning4j.nn.conf.WorkspaceMode trainingWorkspaceMode

inferenceWorkspaceMode

protected org.deeplearning4j.nn.conf.WorkspaceMode inferenceWorkspaceMode

Constructor Detail

Builder

public Builder()

Method Detail

earlyStoppingConfiguration

public ComputationGraphSpace.Builder earlyStoppingConfiguration(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<org.deeplearning4j.nn.graph.ComputationGraph> earlyStoppingConfiguration)

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

layer

public ComputationGraphSpace.Builder layer(String layerName,
                                           LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace,
                                           String... layerInputs)

layer

public ComputationGraphSpace.Builder layer(String layerName,
                                           LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace,
                                           org.deeplearning4j.nn.conf.InputPreProcessor preProcessor,
                                           String... layerInputs)

layer

public ComputationGraphSpace.Builder layer(String layerName,
                                           org.deeplearning4j.nn.conf.layers.Layer layer,
                                           String... layerInputs)

addLayer

public ComputationGraphSpace.Builder addLayer(String layerName,
                                              LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace,
                                              String... layerInputs)

addLayer

public ComputationGraphSpace.Builder addLayer(String layerName,
                                              LayerSpace<? extends org.deeplearning4j.nn.conf.layers.Layer> layerSpace,
                                              org.deeplearning4j.nn.conf.InputPreProcessor preProcessor,
                                              String... layerInputs)

addVertex

public ComputationGraphSpace.Builder addVertex(String vertexName,
                                               org.deeplearning4j.nn.conf.graph.GraphVertex vertex,
                                               String... vertexInputs)

addInputs

public ComputationGraphSpace.Builder addInputs(String... networkInputs)

setOutputs

public ComputationGraphSpace.Builder setOutputs(String... networkOutputs)

setInputTypes

public ComputationGraphSpace.Builder setInputTypes(org.deeplearning4j.nn.conf.inputs.InputType... inputTypes)

setInputTypes

public ComputationGraphSpace.Builder setInputTypes(ParameterSpace<org.deeplearning4j.nn.conf.inputs.InputType[]> inputTypes)

trainingWorkspaceMode

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

inferenceWorkspaceMode

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

build

public ComputationGraphSpace build()

Specified by:

build in class BaseNetworkSpace.Builder<ComputationGraphSpace.Builder>