org.nd4j.autodiff.samediff.config

Class FitConfig

java.lang.Object

org.nd4j.autodiff.samediff.config.FitConfig

public class FitConfig
extends Object

Constructor Summary

Constructors

Constructor and Description
FitConfig(@NonNull SameDiff sd)

Method Summary

Modifier and Type	Method and Description
FitConfig	epochs(int epochs) Set the number of epochs to train for
History	exec() Do the training.
FitConfig	listeners(Listener... listeners) Add listeners for this operation
FitConfig	train(@NonNull DataSetIterator trainingData) Set the training data
FitConfig	train(@NonNull DataSetIterator trainingData, int epochs) Set the training data and number of epochs
FitConfig	train(@NonNull MultiDataSetIterator trainingData) Set the training data
FitConfig	train(@NonNull MultiDataSetIterator trainingData, int epochs) Set the training data and number of epochs
FitConfig	validate(DataSetIterator validationData) Set the validation data
FitConfig	validate(DataSetIterator validationData, int validationFrequency) Set the validation data and frequency
FitConfig	validate(MultiDataSetIterator validationData) Set the validation data
FitConfig	validate(MultiDataSetIterator validationData, int validationFrequency) Set the validation data and frequency
FitConfig	validationFrequency(int validationFrequency) Set the validation frequency.

Methods inherited from class java.lang.Object

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

Constructor Detail

FitConfig

public FitConfig(@NonNull
                 @NonNull SameDiff sd)

Method Detail

epochs

public FitConfig epochs(int epochs)

Set the number of epochs to train for

train

public FitConfig train(@NonNull
                       @NonNull MultiDataSetIterator trainingData)

Set the training data

train

public FitConfig train(@NonNull
                       @NonNull DataSetIterator trainingData)

Set the training data

train

public FitConfig train(@NonNull
                       @NonNull MultiDataSetIterator trainingData,
                       int epochs)

Set the training data and number of epochs

train

public FitConfig train(@NonNull
                       @NonNull DataSetIterator trainingData,
                       int epochs)

Set the training data and number of epochs

validate

public FitConfig validate(MultiDataSetIterator validationData)

Set the validation data

validate

public FitConfig validate(DataSetIterator validationData)

Set the validation data

validationFrequency

public FitConfig validationFrequency(int validationFrequency)

Set the validation frequency. Validation will be preformed once every so many epochs.

Specifically, validation will be preformed when i % validationFrequency == 0

validate

public FitConfig validate(MultiDataSetIterator validationData,
                          int validationFrequency)

Set the validation data and frequency

Specifically, validation will be preformed when i % validationFrequency == 0

validate

public FitConfig validate(DataSetIterator validationData,
                          int validationFrequency)

Set the validation data and frequency

Specifically, validation will be preformed when i % validationFrequency == 0

listeners

public FitConfig listeners(@NonNull
                           Listener... listeners)

Add listeners for this operation

exec

public History exec()

Do the training.

Returns:

a History object containing the history information for this training operation (evaluations specified in the TrainingConfig, loss values, and timing information).