org.deeplearning4j.nn.conf

Class NeuralNetConfiguration.Builder

java.lang.Object

org.deeplearning4j.nn.conf.NeuralNetConfiguration.Builder

Enclosing class:

NeuralNetConfiguration

public static class NeuralNetConfiguration.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
NeuralNetConfiguration.Builder	activationFunction(org.nd4j.linalg.api.activation.ActivationFunction activationFunction)
NeuralNetConfiguration.Builder	adagradResetIterations(int resetAdaGradIterations)
NeuralNetConfiguration.Builder	applySparsity(boolean applySparsity)
NeuralNetConfiguration.Builder	batchSize(int batchSize)
NeuralNetConfiguration	build()
NeuralNetConfiguration.Builder	clone()
NeuralNetConfiguration.Builder	concatBiases(boolean concatBiases)
NeuralNetConfiguration.Builder	constrainGradientToUnitNorm(boolean constrainGradientToUnitNorm)
NeuralNetConfiguration.Builder	corruptionLevel(double corruptionLevel)
NeuralNetConfiguration.Builder	dist(org.apache.commons.math3.distribution.RealDistribution dist)
NeuralNetConfiguration.Builder	dropOut(double dropOut)
NeuralNetConfiguration.Builder	featureMapSize(int[] featureMapSize)
NeuralNetConfiguration.Builder	filterSize(int... filterSize)
NeuralNetConfiguration.Builder	hiddenUnit(RBM.HiddenUnit hiddenUnit)
NeuralNetConfiguration.Builder	iterationListener(IterationListener listener)
NeuralNetConfiguration.Builder	iterations(int numIterations)
NeuralNetConfiguration.Builder	k(int k)
NeuralNetConfiguration.Builder	kernel(int kernel)
NeuralNetConfiguration.Builder	l2(double l2)
NeuralNetConfiguration.Builder	layerFactory(LayerFactory layerFactory)
NeuralNetConfiguration.Builder	learningRate(double lr)
NeuralNetConfiguration.ListBuilder	list(int size)
NeuralNetConfiguration.Builder	listeners(Collection<IterationListener> listeners)
NeuralNetConfiguration.Builder	lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
NeuralNetConfiguration.Builder	momentum(double momentum)
NeuralNetConfiguration.Builder	momentumAfter(Map<Integer,Double> momentumAfter)
NeuralNetConfiguration.Builder	nIn(int nIn)
NeuralNetConfiguration.Builder	nOut(int nOut)
NeuralNetConfiguration.Builder	optimizationAlgo(OptimizationAlgorithm optimizationAlgo)
NeuralNetConfiguration.Builder	regularization(boolean useRegularization)
NeuralNetConfiguration.Builder	render(int renderWeightsEveryNumEpochs)
NeuralNetConfiguration.Builder	resetAdaGradIterations(int resetAdaGradIterations)
NeuralNetConfiguration.Builder	rng(org.apache.commons.math3.random.RandomGenerator rng)
NeuralNetConfiguration.Builder	seed(long seed)
NeuralNetConfiguration.Builder	sparsity(double sparsity)
NeuralNetConfiguration.Builder	stepFunction(StepFunction stepFunction)
NeuralNetConfiguration.Builder	stride(int[] stride)
NeuralNetConfiguration.Builder	useAdaGrad(boolean useAdaGrad)
NeuralNetConfiguration.Builder	visibleUnit(RBM.VisibleUnit visibleUnit)
NeuralNetConfiguration.Builder	weightInit(WeightInit weightInit)
NeuralNetConfiguration.Builder	weightShape(int[] weightShape)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

batchSize

public NeuralNetConfiguration.Builder batchSize(int batchSize)

kernel

public NeuralNetConfiguration.Builder kernel(int kernel)

layerFactory

public NeuralNetConfiguration.Builder layerFactory(LayerFactory layerFactory)

stepFunction

public NeuralNetConfiguration.Builder stepFunction(StepFunction stepFunction)

list

public NeuralNetConfiguration.ListBuilder list(int size)

clone

public NeuralNetConfiguration.Builder clone()

Overrides:

clone in class Object

iterationListener

public NeuralNetConfiguration.Builder iterationListener(IterationListener listener)

listeners

public NeuralNetConfiguration.Builder listeners(Collection<IterationListener> listeners)

featureMapSize

public NeuralNetConfiguration.Builder featureMapSize(int[] featureMapSize)

stride

public NeuralNetConfiguration.Builder stride(int[] stride)

filterSize

public NeuralNetConfiguration.Builder filterSize(int... filterSize)

weightShape

public NeuralNetConfiguration.Builder weightShape(int[] weightShape)

iterations

public NeuralNetConfiguration.Builder iterations(int numIterations)

dist

public NeuralNetConfiguration.Builder dist(org.apache.commons.math3.distribution.RealDistribution dist)

sparsity

public NeuralNetConfiguration.Builder sparsity(double sparsity)

useAdaGrad

public NeuralNetConfiguration.Builder useAdaGrad(boolean useAdaGrad)

learningRate

public NeuralNetConfiguration.Builder learningRate(double lr)

momentum

public NeuralNetConfiguration.Builder momentum(double momentum)

k

public NeuralNetConfiguration.Builder k(int k)

corruptionLevel

public NeuralNetConfiguration.Builder corruptionLevel(double corruptionLevel)

momentumAfter

public NeuralNetConfiguration.Builder momentumAfter(Map<Integer,Double> momentumAfter)

adagradResetIterations

public NeuralNetConfiguration.Builder adagradResetIterations(int resetAdaGradIterations)

dropOut

public NeuralNetConfiguration.Builder dropOut(double dropOut)

applySparsity

public NeuralNetConfiguration.Builder applySparsity(boolean applySparsity)

weightInit

public NeuralNetConfiguration.Builder weightInit(WeightInit weightInit)

render

public NeuralNetConfiguration.Builder render(int renderWeightsEveryNumEpochs)

concatBiases

public NeuralNetConfiguration.Builder concatBiases(boolean concatBiases)

rng

public NeuralNetConfiguration.Builder rng(org.apache.commons.math3.random.RandomGenerator rng)

seed

public NeuralNetConfiguration.Builder seed(long seed)

build

public NeuralNetConfiguration build()

l2

public NeuralNetConfiguration.Builder l2(double l2)

regularization

public NeuralNetConfiguration.Builder regularization(boolean useRegularization)

resetAdaGradIterations

public NeuralNetConfiguration.Builder resetAdaGradIterations(int resetAdaGradIterations)

optimizationAlgo

public NeuralNetConfiguration.Builder optimizationAlgo(OptimizationAlgorithm optimizationAlgo)

lossFunction

public NeuralNetConfiguration.Builder lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

constrainGradientToUnitNorm

public NeuralNetConfiguration.Builder constrainGradientToUnitNorm(boolean constrainGradientToUnitNorm)

nIn

public NeuralNetConfiguration.Builder nIn(int nIn)

nOut

public NeuralNetConfiguration.Builder nOut(int nOut)

activationFunction

public NeuralNetConfiguration.Builder activationFunction(org.nd4j.linalg.api.activation.ActivationFunction activationFunction)

visibleUnit

public NeuralNetConfiguration.Builder visibleUnit(RBM.VisibleUnit visibleUnit)

hiddenUnit

public NeuralNetConfiguration.Builder hiddenUnit(RBM.HiddenUnit hiddenUnit)