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	build()
NeuralNetConfiguration.Builder	clone()
NeuralNetConfiguration.Builder	concatBiases(boolean concatBiases)
NeuralNetConfiguration.Builder	constrainGradientToUnitNorm(boolean constrainGradientToUnitNorm)
NeuralNetConfiguration.Builder	dist(org.apache.commons.math3.distribution.RealDistribution dist)
NeuralNetConfiguration.Builder	dropOut(float dropOut)
NeuralNetConfiguration.Builder	featureMapSize(int[] featureMapSize)
NeuralNetConfiguration.Builder	filterSize(int[] filterSize)
NeuralNetConfiguration.Builder	hiddenUnit(RBM.HiddenUnit hiddenUnit)
NeuralNetConfiguration.Builder	iterations(int numIterations)
NeuralNetConfiguration.Builder	l2(float l2)
NeuralNetConfiguration.Builder	learningRate(float lr)
NeuralNetConfiguration.ListBuilder	list(int size)
NeuralNetConfiguration.Builder	lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
NeuralNetConfiguration.Builder	momentum(float momentum)
NeuralNetConfiguration.Builder	momentumAfter(Map<Integer,Float> momentumAfter)
NeuralNetConfiguration.Builder	nIn(int nIn)
NeuralNetConfiguration.Builder	nOut(int nOut)
NeuralNetConfiguration.Builder	numFeatureMaps(int numFeatureMaps)
NeuralNetConfiguration.Builder	numInFeatureMaps(int numInFeatureMaps)
NeuralNetConfiguration.Builder	optimizationAlgo(NeuralNetwork.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(float sparsity)
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)
NeuralNetConfiguration.Builder	withActivationType(NeuralNetConfiguration.ActivationType activationType)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

list

public NeuralNetConfiguration.ListBuilder list(int size)

clone

public NeuralNetConfiguration.Builder clone()

Overrides:

clone in class Object

numInFeatureMaps

public NeuralNetConfiguration.Builder numInFeatureMaps(int numInFeatureMaps)

featureMapSize

public NeuralNetConfiguration.Builder featureMapSize(int[] featureMapSize)

stride

public NeuralNetConfiguration.Builder stride(int[] stride)

numFeatureMaps

public NeuralNetConfiguration.Builder numFeatureMaps(int numFeatureMaps)

filterSize

public NeuralNetConfiguration.Builder filterSize(int[] filterSize)

weightShape

public NeuralNetConfiguration.Builder weightShape(int[] weightShape)

withActivationType

public NeuralNetConfiguration.Builder withActivationType(NeuralNetConfiguration.ActivationType activationType)

iterations

public NeuralNetConfiguration.Builder iterations(int numIterations)

dist

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

sparsity

public NeuralNetConfiguration.Builder sparsity(float sparsity)

useAdaGrad

public NeuralNetConfiguration.Builder useAdaGrad(boolean useAdaGrad)

learningRate

public NeuralNetConfiguration.Builder learningRate(float lr)

momentum

public NeuralNetConfiguration.Builder momentum(float momentum)

momentumAfter

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

adagradResetIterations

public NeuralNetConfiguration.Builder adagradResetIterations(int resetAdaGradIterations)

dropOut

public NeuralNetConfiguration.Builder dropOut(float 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(float l2)

regularization

public NeuralNetConfiguration.Builder regularization(boolean useRegularization)

resetAdaGradIterations

public NeuralNetConfiguration.Builder resetAdaGradIterations(int resetAdaGradIterations)

optimizationAlgo

public NeuralNetConfiguration.Builder optimizationAlgo(NeuralNetwork.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)