org.deeplearning4j.nn

Class BaseNeuralNetwork.Builder<E extends BaseNeuralNetwork>

java.lang.Object

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

Direct Known Subclasses:

CRBM.Builder, DenoisingAutoEncoder.Builder, RBM.Builder

Enclosing class:

BaseNeuralNetwork

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

Field Summary

Fields

Modifier and Type	Field and Description
protected Class<? extends NeuralNetwork>	clazz

Constructor Summary

Constructors

Constructor and Description
BaseNeuralNetwork.Builder()

Method Summary

Methods

Modifier and Type	Method and Description
BaseNeuralNetwork.Builder<E>	asType(Class<E> clazz)
E	build()
E	buildEmpty()
BaseNeuralNetwork.Builder<E>	fanIn(double fanIn)
BaseNeuralNetwork.Builder<E>	numberOfVisible(int numVisible)
BaseNeuralNetwork.Builder<E>	numHidden(int numHidden)
BaseNeuralNetwork.Builder<E>	renderWeights(int numEpochs)
BaseNeuralNetwork.Builder<E>	useRegularization(boolean useRegularization)
BaseNeuralNetwork.Builder<E>	withClazz(Class<? extends BaseNeuralNetwork> clazz)
BaseNeuralNetwork.Builder<E>	withDistribution(org.apache.commons.math3.distribution.RealDistribution dist)
BaseNeuralNetwork.Builder<E>	withHBias(org.jblas.DoubleMatrix hBias)
BaseNeuralNetwork.Builder<E>	withInput(org.jblas.DoubleMatrix input)
BaseNeuralNetwork.Builder<E>	withL2(double l2)
BaseNeuralNetwork.Builder<E>	withMomentum(double momentum)
BaseNeuralNetwork.Builder<E>	withRandom(org.apache.commons.math3.random.RandomGenerator gen)
BaseNeuralNetwork.Builder<E>	withSparsity(double sparsity)
BaseNeuralNetwork.Builder<E>	withVisibleBias(org.jblas.DoubleMatrix vBias)
BaseNeuralNetwork.Builder<E>	withWeights(org.jblas.DoubleMatrix W)

Methods inherited from class java.lang.Object

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

Field Detail

clazz

protected Class<? extends NeuralNetwork> clazz

Constructor Detail

BaseNeuralNetwork.Builder

public BaseNeuralNetwork.Builder()

Method Detail

withDistribution

public BaseNeuralNetwork.Builder<E> withDistribution(org.apache.commons.math3.distribution.RealDistribution dist)

useRegularization

public BaseNeuralNetwork.Builder<E> useRegularization(boolean useRegularization)

fanIn

public BaseNeuralNetwork.Builder<E> fanIn(double fanIn)

withL2

public BaseNeuralNetwork.Builder<E> withL2(double l2)

renderWeights

public BaseNeuralNetwork.Builder<E> renderWeights(int numEpochs)

buildEmpty

public E buildEmpty()

withClazz

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

withSparsity

public BaseNeuralNetwork.Builder<E> withSparsity(double sparsity)

withMomentum

public BaseNeuralNetwork.Builder<E> withMomentum(double momentum)

withInput

public BaseNeuralNetwork.Builder<E> withInput(org.jblas.DoubleMatrix input)

asType

public BaseNeuralNetwork.Builder<E> asType(Class<E> clazz)

withWeights

public BaseNeuralNetwork.Builder<E> withWeights(org.jblas.DoubleMatrix W)

withVisibleBias

public BaseNeuralNetwork.Builder<E> withVisibleBias(org.jblas.DoubleMatrix vBias)

withHBias

public BaseNeuralNetwork.Builder<E> withHBias(org.jblas.DoubleMatrix hBias)

numberOfVisible

public BaseNeuralNetwork.Builder<E> numberOfVisible(int numVisible)

numHidden

public BaseNeuralNetwork.Builder<E> numHidden(int numHidden)

withRandom

public BaseNeuralNetwork.Builder<E> withRandom(org.apache.commons.math3.random.RandomGenerator gen)

build

public E build()