org.deeplearning4j.nn.transferlearning

Class FineTuneConfiguration

java.lang.Object

org.deeplearning4j.nn.transferlearning.FineTuneConfiguration

public class FineTuneConfiguration
extends java.lang.Object

Created by Alex on 21/02/2017.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	FineTuneConfiguration.Builder

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.activations.IActivation	activationFn
protected java.lang.Double	adamMeanDecay
protected java.lang.Double	adamVarDecay
protected java.lang.Boolean	backprop
protected BackpropType	backpropType
protected java.lang.Double	biasInit
protected java.lang.Double	biasLearningRate
protected ConvolutionMode	convolutionMode
protected Distribution	dist
protected java.lang.Double	dropOut
protected java.lang.Double	epsilon
protected GradientNormalization	gradientNormalization
protected java.lang.Double	gradientNormalizationThreshold
protected java.lang.Double	l1
protected java.lang.Double	l1Bias
protected java.lang.Double	l2
protected java.lang.Double	l2Bias
protected java.lang.Double	learningRate
protected LearningRatePolicy	learningRatePolicy
protected java.util.Map<java.lang.Integer,java.lang.Double>	learningRateSchedule
protected java.lang.Double	lrPolicyDecayRate
protected java.lang.Double	lrPolicyPower
protected java.lang.Double	lrPolicySteps
protected java.lang.Double	lrScoreBasedDecay
protected java.lang.Integer	maxNumLineSearchIterations
protected java.lang.Boolean	miniBatch
protected java.lang.Boolean	minimize
protected java.lang.Double	momentum
protected java.util.Map<java.lang.Integer,java.lang.Double>	momentumSchedule
protected java.lang.Integer	numIterations
protected OptimizationAlgorithm	optimizationAlgo
protected java.lang.Boolean	pretrain
protected java.lang.Double	rho
protected java.lang.Double	rmsDecay
protected java.lang.Long	seed
protected StepFunction	stepFunction
protected java.lang.Integer	tbpttBackLength
protected java.lang.Integer	tbpttFwdLength
protected Updater	updater
protected java.lang.Boolean	useDropConnect
protected java.lang.Boolean	useRegularization
protected WeightInit	weightInit

Constructor Summary

Constructors

Constructor and Description
FineTuneConfiguration()

Method Summary

Modifier and Type	Method and Description
NeuralNetConfiguration	appliedNeuralNetConfiguration(NeuralNetConfiguration nnc)
NeuralNetConfiguration.Builder	appliedNeuralNetConfigurationBuilder()
void	applyToComputationGraphConfiguration(ComputationGraphConfiguration conf)
void	applyToMultiLayerConfiguration(MultiLayerConfiguration conf)
void	applyToNeuralNetConfiguration(NeuralNetConfiguration nnc)
static FineTuneConfiguration	fromJson(java.lang.String json)
static FineTuneConfiguration	fromYaml(java.lang.String yaml)
java.lang.String	toJson()
java.lang.String	toYaml()

Methods inherited from class java.lang.Object

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

Field Detail

activationFn

protected org.nd4j.linalg.activations.IActivation activationFn

weightInit

protected WeightInit weightInit

biasInit

protected java.lang.Double biasInit

dist

protected Distribution dist

learningRate

protected java.lang.Double learningRate

biasLearningRate

protected java.lang.Double biasLearningRate

learningRateSchedule

protected java.util.Map<java.lang.Integer,java.lang.Double> learningRateSchedule

lrScoreBasedDecay

protected java.lang.Double lrScoreBasedDecay

l1

protected java.lang.Double l1

l2

protected java.lang.Double l2

l1Bias

protected java.lang.Double l1Bias

l2Bias

protected java.lang.Double l2Bias

dropOut

protected java.lang.Double dropOut

updater

protected Updater updater

momentum

protected java.lang.Double momentum

momentumSchedule

protected java.util.Map<java.lang.Integer,java.lang.Double> momentumSchedule

epsilon

protected java.lang.Double epsilon

rho

protected java.lang.Double rho

rmsDecay

protected java.lang.Double rmsDecay

adamMeanDecay

protected java.lang.Double adamMeanDecay

adamVarDecay

protected java.lang.Double adamVarDecay

miniBatch

protected java.lang.Boolean miniBatch

numIterations

protected java.lang.Integer numIterations

maxNumLineSearchIterations

protected java.lang.Integer maxNumLineSearchIterations

seed

protected java.lang.Long seed

useRegularization

protected java.lang.Boolean useRegularization

optimizationAlgo

protected OptimizationAlgorithm optimizationAlgo

stepFunction

protected StepFunction stepFunction

useDropConnect

protected java.lang.Boolean useDropConnect

minimize

protected java.lang.Boolean minimize

gradientNormalization

protected GradientNormalization gradientNormalization

gradientNormalizationThreshold

protected java.lang.Double gradientNormalizationThreshold

learningRatePolicy

protected LearningRatePolicy learningRatePolicy

lrPolicyDecayRate

protected java.lang.Double lrPolicyDecayRate

lrPolicySteps

protected java.lang.Double lrPolicySteps

lrPolicyPower

protected java.lang.Double lrPolicyPower

convolutionMode

protected ConvolutionMode convolutionMode

pretrain

protected java.lang.Boolean pretrain

backprop

protected java.lang.Boolean backprop

backpropType

protected BackpropType backpropType

tbpttFwdLength

protected java.lang.Integer tbpttFwdLength

tbpttBackLength

protected java.lang.Integer tbpttBackLength

Constructor Detail

FineTuneConfiguration

public FineTuneConfiguration()

Method Detail

appliedNeuralNetConfiguration

public NeuralNetConfiguration appliedNeuralNetConfiguration(NeuralNetConfiguration nnc)

applyToNeuralNetConfiguration

public void applyToNeuralNetConfiguration(NeuralNetConfiguration nnc)

applyToMultiLayerConfiguration

public void applyToMultiLayerConfiguration(MultiLayerConfiguration conf)

applyToComputationGraphConfiguration

public void applyToComputationGraphConfiguration(ComputationGraphConfiguration conf)

appliedNeuralNetConfigurationBuilder

public NeuralNetConfiguration.Builder appliedNeuralNetConfigurationBuilder()

toJson

public java.lang.String toJson()

toYaml

public java.lang.String toYaml()

fromJson

public static FineTuneConfiguration fromJson(java.lang.String json)

fromYaml

public static FineTuneConfiguration fromYaml(java.lang.String yaml)