org.deeplearning4j.nn.conf

Class NeuralNetConfiguration.ListBuilder

java.lang.Object

org.deeplearning4j.nn.conf.MultiLayerConfiguration.Builder

org.deeplearning4j.nn.conf.NeuralNetConfiguration.ListBuilder

Enclosing class:

NeuralNetConfiguration

public static class NeuralNetConfiguration.ListBuilder
extends MultiLayerConfiguration.Builder

Fluent interface for building a list of configurations

Field Summary

Fields inherited from class org.deeplearning4j.nn.conf.MultiLayerConfiguration.Builder

backprop, backpropType, cnnInputSize, confs, dampingFactor, inputPreProcessors, inputType, pretrain, tbpttBackLength, tbpttFwdLength

Constructor Summary

Constructors

Constructor and Description
ListBuilder(NeuralNetConfiguration.Builder globalConfig)
ListBuilder(NeuralNetConfiguration.Builder globalConfig, java.util.Map<java.lang.Integer,NeuralNetConfiguration.Builder> layerMap)

Method Summary

Modifier and Type	Method and Description
NeuralNetConfiguration.ListBuilder	backprop(boolean backprop) Whether to do back prop or not
MultiLayerConfiguration	build() Build the multi layer network based on this neural network and overr ridden parameters
java.util.Map<java.lang.Integer,NeuralNetConfiguration.Builder>	getLayerwise()
NeuralNetConfiguration.ListBuilder	layer(int ind, Layer layer)
NeuralNetConfiguration.ListBuilder	pretrain(boolean pretrain) Whether to do pre train or not

Methods inherited from class org.deeplearning4j.nn.conf.MultiLayerConfiguration.Builder

backpropType, cnnInputSize, cnnInputSize, confs, inputPreProcessor, inputPreProcessors, setInputType, tBPTTBackwardLength, tBPTTForwardLength

Methods inherited from class java.lang.Object

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

Constructor Detail

ListBuilder

public ListBuilder(NeuralNetConfiguration.Builder globalConfig,
                   java.util.Map<java.lang.Integer,NeuralNetConfiguration.Builder> layerMap)

ListBuilder

public ListBuilder(NeuralNetConfiguration.Builder globalConfig)

Method Detail

backprop

public NeuralNetConfiguration.ListBuilder backprop(boolean backprop)

Description copied from class: MultiLayerConfiguration.Builder

Whether to do back prop or not

Overrides:

backprop in class MultiLayerConfiguration.Builder

Parameters:

backprop - whether to do back prop or not

Returns:

pretrain

public NeuralNetConfiguration.ListBuilder pretrain(boolean pretrain)

Description copied from class: MultiLayerConfiguration.Builder

Whether to do pre train or not

Overrides:

pretrain in class MultiLayerConfiguration.Builder

Parameters:

pretrain - whether to do pre train or not

Returns:

builder pattern

layer

public NeuralNetConfiguration.ListBuilder layer(int ind,
                                                Layer layer)

getLayerwise

public java.util.Map<java.lang.Integer,NeuralNetConfiguration.Builder> getLayerwise()

build

public MultiLayerConfiguration build()

Build the multi layer network based on this neural network and overr ridden parameters

Overrides:

build in class MultiLayerConfiguration.Builder

Returns:

the configuration to build