org.deeplearning4j.nn

Class BaseNeuralNetwork.Builder<E extends BaseNeuralNetwork>

java.lang.Object

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

Direct Known Subclasses:

AutoEncoder.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
Builder()

Method Summary

Modifier and Type	Method and Description
BaseNeuralNetwork.Builder<E>	asType(Class<E> clazz)
E	build()
E	buildEmpty()
BaseNeuralNetwork.Builder<E>	configure(NeuralNetConfiguration conf)
BaseNeuralNetwork.Builder<E>	withClazz(Class<? extends BaseNeuralNetwork> clazz)
BaseNeuralNetwork.Builder<E>	withHBias(org.nd4j.linalg.api.ndarray.INDArray hBias)
BaseNeuralNetwork.Builder<E>	withInput(org.nd4j.linalg.api.ndarray.INDArray input)
BaseNeuralNetwork.Builder<E>	withVisibleBias(org.nd4j.linalg.api.ndarray.INDArray vBias)
BaseNeuralNetwork.Builder<E>	withWeights(org.nd4j.linalg.api.ndarray.INDArray 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

Builder

public Builder()

Method Detail

configure

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

buildEmpty

public E buildEmpty()

withClazz

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

withInput

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

asType

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

withWeights

public BaseNeuralNetwork.Builder<E> withWeights(org.nd4j.linalg.api.ndarray.INDArray W)

withVisibleBias

public BaseNeuralNetwork.Builder<E> withVisibleBias(org.nd4j.linalg.api.ndarray.INDArray vBias)

withHBias

public BaseNeuralNetwork.Builder<E> withHBias(org.nd4j.linalg.api.ndarray.INDArray hBias)

build

public E build()