org.deeplearning4j.nn.transferlearning

Class FineTuneConfiguration.Builder

java.lang.Object

org.deeplearning4j.nn.transferlearning.FineTuneConfiguration.Builder

Enclosing class:

FineTuneConfiguration

public static class FineTuneConfiguration.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
FineTuneConfiguration.Builder	activation(org.nd4j.linalg.activations.Activation activation)
FineTuneConfiguration.Builder	iterations(int iterations)
FineTuneConfiguration.Builder	regularization(boolean regularization)
FineTuneConfiguration.Builder	seed(int seed)
FineTuneConfiguration.Builder	seed(long seed)
FineTuneConfiguration.Builder	updater(org.nd4j.linalg.learning.config.IUpdater updater)
FineTuneConfiguration.Builder	updater(Updater updater)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

seed

public FineTuneConfiguration.Builder seed(int seed)

seed

public FineTuneConfiguration.Builder seed(long seed)

regularization

public FineTuneConfiguration.Builder regularization(boolean regularization)

iterations

public FineTuneConfiguration.Builder iterations(int iterations)

activation

public FineTuneConfiguration.Builder activation(org.nd4j.linalg.activations.Activation activation)

updater

public FineTuneConfiguration.Builder updater(org.nd4j.linalg.learning.config.IUpdater updater)

updater

public FineTuneConfiguration.Builder updater(Updater updater)