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(String activationFunction)
NeuralNetConfiguration.Builder	adagradResetIterations(int resetAdaGradIterations) Deprecated.
NeuralNetConfiguration.Builder	applySparsity(boolean applySparsity)
NeuralNetConfiguration.Builder	batchSize(int batchSize)
NeuralNetConfiguration	build()
NeuralNetConfiguration.Builder	clone()
NeuralNetConfiguration.Builder	constrainGradientToUnitNorm(boolean constrainGradientToUnitNorm)
NeuralNetConfiguration.Builder	convolutionType(ConvolutionLayer.ConvolutionType convolutionType)
NeuralNetConfiguration.Builder	corruptionLevel(double corruptionLevel)
NeuralNetConfiguration.Builder	customLossFunction(String customLossFunction)
NeuralNetConfiguration.Builder	dist(Distribution dist)
NeuralNetConfiguration.Builder	dropOut(double dropOut)
NeuralNetConfiguration.Builder	featureMapSize(int... featureMapSize)
NeuralNetConfiguration.Builder	filterSize(int... filterSize)
NeuralNetConfiguration.Builder	hiddenUnit(RBM.HiddenUnit hiddenUnit)
NeuralNetConfiguration.Builder	iterations(int numIterations)
NeuralNetConfiguration.Builder	k(int k)
NeuralNetConfiguration.Builder	kernel(int kernel)
NeuralNetConfiguration.Builder	l1(double l1)
NeuralNetConfiguration.Builder	l2(double l2)
NeuralNetConfiguration.Builder	layer(Layer layer)
NeuralNetConfiguration.Builder	learningRate(double lr)
NeuralNetConfiguration.ListBuilder	list(int size)
NeuralNetConfiguration.Builder	lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
NeuralNetConfiguration.Builder	minimize(boolean minimize)
NeuralNetConfiguration.Builder	momentum(double momentum)
NeuralNetConfiguration.Builder	momentumAfter(Map<Integer,Double> momentumAfter)
NeuralNetConfiguration.Builder	nIn(int nIn)
NeuralNetConfiguration.Builder	nOut(int nOut)
NeuralNetConfiguration.Builder	numLineSearchIterations(int numLineSearchIterations)
NeuralNetConfiguration.Builder	optimizationAlgo(OptimizationAlgorithm optimizationAlgo)
NeuralNetConfiguration.Builder	regularization(boolean useRegularization)
NeuralNetConfiguration.Builder	resetAdaGradIterations(int resetAdaGradIterations)
NeuralNetConfiguration.Builder	rng(Random rng)
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

l1

public NeuralNetConfiguration.Builder l1(double l1)

customLossFunction

public NeuralNetConfiguration.Builder customLossFunction(String customLossFunction)

convolutionType

public NeuralNetConfiguration.Builder convolutionType(ConvolutionLayer.ConvolutionType convolutionType)

minimize

public NeuralNetConfiguration.Builder minimize(boolean minimize)

numLineSearchIterations

public NeuralNetConfiguration.Builder numLineSearchIterations(int numLineSearchIterations)

batchSize

public NeuralNetConfiguration.Builder batchSize(int batchSize)

kernel

public NeuralNetConfiguration.Builder kernel(int kernel)

layer

public NeuralNetConfiguration.Builder layer(Layer layer)

stepFunction

public NeuralNetConfiguration.Builder stepFunction(StepFunction stepFunction)

list

public NeuralNetConfiguration.ListBuilder list(int size)

clone

public NeuralNetConfiguration.Builder clone()

Overrides:

clone in class Object

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(Distribution 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

@Deprecated
public NeuralNetConfiguration.Builder adagradResetIterations(int resetAdaGradIterations)

Deprecated.

dropOut

public NeuralNetConfiguration.Builder dropOut(double dropOut)

applySparsity

public NeuralNetConfiguration.Builder applySparsity(boolean applySparsity)

weightInit

public NeuralNetConfiguration.Builder weightInit(WeightInit weightInit)

rng

public NeuralNetConfiguration.Builder rng(Random rng)

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(String activationFunction)

visibleUnit

public NeuralNetConfiguration.Builder visibleUnit(RBM.VisibleUnit visibleUnit)

hiddenUnit

public NeuralNetConfiguration.Builder hiddenUnit(RBM.HiddenUnit hiddenUnit)