org.deeplearning4j.nn

Class BaseMultiLayerNetwork.Builder<E extends BaseMultiLayerNetwork>

java.lang.Object

org.deeplearning4j.nn.BaseMultiLayerNetwork.Builder<E>

Direct Known Subclasses:

DBN.Builder, SemanticHashing.Builder, StackedDenoisingAutoEncoder.Builder

Enclosing class:

BaseMultiLayerNetwork

public static class BaseMultiLayerNetwork.Builder<E extends BaseMultiLayerNetwork>
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	backProp
protected Class<? extends BaseMultiLayerNetwork>	clazz
protected NeuralNetConfiguration	conf
protected List<NeuralNetConfiguration>	layerWiseConfiguration
protected boolean	shouldForceEpochs
protected Map<Integer,org.nd4j.linalg.transformation.MatrixTransform>	weightTransforms

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
E	build()
E	buildEmpty()
BaseMultiLayerNetwork.Builder<E>	configure(NeuralNetConfiguration conf)
BaseMultiLayerNetwork.Builder<E>	disableBackProp() Disables back propagation
BaseMultiLayerNetwork.Builder<E>	forceEpochs() Forces use of number of epochs for training SGD style rather than conjugate gradient
BaseMultiLayerNetwork.Builder<E>	hiddenLayerSizes(int[] hiddenLayerSizes)
BaseMultiLayerNetwork.Builder<E>	hiddenLayerSizes(Integer[] hiddenLayerSizes) Size of the hidden neuralNets
BaseMultiLayerNetwork.Builder<E>	layerWiseCOnfiguration(List<NeuralNetConfiguration> layerWiseConfiguration)
BaseMultiLayerNetwork.Builder<E>	lineSearchBackProp(boolean lineSearchBackProp) Whether to use line search back prop instead of SGD
BaseMultiLayerNetwork.Builder<E>	transformWeightsAt(int layer, org.nd4j.linalg.transformation.MatrixTransform transform) Transform the weights at the given layer
BaseMultiLayerNetwork.Builder<E>	transformWeightsAt(Map<Integer,org.nd4j.linalg.transformation.MatrixTransform> transforms) A map of transformations for transforming the given neuralNets
BaseMultiLayerNetwork.Builder<E>	useDropConnection(boolean useDropConnect) Use drop connect on activations or not
BaseMultiLayerNetwork.Builder<E>	useGaussNewtonVectorProductBackProp(boolean useGaussNewtonVectorProductBackProp) Use gauss newton back prop - this is for hessian free
BaseMultiLayerNetwork.Builder<E>	withClazz(Class<? extends BaseMultiLayerNetwork> clazz)
BaseMultiLayerNetwork.Builder<E>	withHiddenBiasTransforms(Map<Integer,org.nd4j.linalg.transformation.MatrixTransform> hiddenBiasTransforms)
BaseMultiLayerNetwork.Builder<E>	withInput(org.nd4j.linalg.api.ndarray.INDArray input)
BaseMultiLayerNetwork.Builder<E>	withLabels(org.nd4j.linalg.api.ndarray.INDArray labels)
BaseMultiLayerNetwork.Builder<E>	withVisibleBiasTransforms(Map<Integer,org.nd4j.linalg.transformation.MatrixTransform> visibleBiasTransforms)

Methods inherited from class java.lang.Object

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

Field Detail

clazz

protected Class<? extends BaseMultiLayerNetwork> clazz

weightTransforms

protected Map<Integer,org.nd4j.linalg.transformation.MatrixTransform> weightTransforms

backProp

protected boolean backProp

shouldForceEpochs

protected boolean shouldForceEpochs

conf

protected NeuralNetConfiguration conf

layerWiseConfiguration

protected List<NeuralNetConfiguration> layerWiseConfiguration

Constructor Detail

Builder

public Builder()

Method Detail

layerWiseCOnfiguration

public BaseMultiLayerNetwork.Builder<E> layerWiseCOnfiguration(List<NeuralNetConfiguration> layerWiseConfiguration)

configure

public BaseMultiLayerNetwork.Builder<E> configure(NeuralNetConfiguration conf)

useGaussNewtonVectorProductBackProp

public BaseMultiLayerNetwork.Builder<E> useGaussNewtonVectorProductBackProp(boolean useGaussNewtonVectorProductBackProp)

Use gauss newton back prop - this is for hessian free

Parameters:

useGaussNewtonVectorProductBackProp - whether to use gauss newton vector backprop

Returns:

useDropConnection

public BaseMultiLayerNetwork.Builder<E> useDropConnection(boolean useDropConnect)

Use drop connect on activations or not

Parameters:

useDropConnect - use drop connect or not

Returns:

builder pattern

lineSearchBackProp

public BaseMultiLayerNetwork.Builder<E> lineSearchBackProp(boolean lineSearchBackProp)

Whether to use line search back prop instead of SGD

Parameters:

lineSearchBackProp -

Returns:

withVisibleBiasTransforms

public BaseMultiLayerNetwork.Builder<E> withVisibleBiasTransforms(Map<Integer,org.nd4j.linalg.transformation.MatrixTransform> visibleBiasTransforms)

withHiddenBiasTransforms

public BaseMultiLayerNetwork.Builder<E> withHiddenBiasTransforms(Map<Integer,org.nd4j.linalg.transformation.MatrixTransform> hiddenBiasTransforms)

forceEpochs

public BaseMultiLayerNetwork.Builder<E> forceEpochs()

Forces use of number of epochs for training SGD style rather than conjugate gradient

Returns:

disableBackProp

public BaseMultiLayerNetwork.Builder<E> disableBackProp()

Disables back propagation

Returns:

transformWeightsAt

public BaseMultiLayerNetwork.Builder<E> transformWeightsAt(int layer,
                                                           org.nd4j.linalg.transformation.MatrixTransform transform)

Transform the weights at the given layer

Parameters:

layer - the layer to applyTransformToOrigin

transform - the function used for transformation

Returns:

transformWeightsAt

public BaseMultiLayerNetwork.Builder<E> transformWeightsAt(Map<Integer,org.nd4j.linalg.transformation.MatrixTransform> transforms)

A map of transformations for transforming the given neuralNets

Parameters:

transforms -

Returns:

hiddenLayerSizes

public BaseMultiLayerNetwork.Builder<E> hiddenLayerSizes(Integer[] hiddenLayerSizes)

Size of the hidden neuralNets

Parameters:

hiddenLayerSizes -

Returns:

hiddenLayerSizes

public BaseMultiLayerNetwork.Builder<E> hiddenLayerSizes(int[] hiddenLayerSizes)

withInput

public BaseMultiLayerNetwork.Builder<E> withInput(org.nd4j.linalg.api.ndarray.INDArray input)

withLabels

public BaseMultiLayerNetwork.Builder<E> withLabels(org.nd4j.linalg.api.ndarray.INDArray labels)

withClazz

public BaseMultiLayerNetwork.Builder<E> withClazz(Class<? extends BaseMultiLayerNetwork> clazz)

buildEmpty

public E buildEmpty()

build

public E build()