Class FitConfig
- java.lang.Object
-
- org.nd4j.autodiff.samediff.config.FitConfig
-
public class FitConfig extends Object
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description FitConfig
epochs(int epochs)
Set the number of epochs to train forHistory
exec()
Do the training.FitConfig
listeners(@NonNull Listener... listeners)
Add listeners for this operationFitConfig
train(@NonNull DataSetIterator trainingData)
Set the training dataFitConfig
train(@NonNull DataSetIterator trainingData, int epochs)
Set the training data and number of epochsFitConfig
train(@NonNull MultiDataSetIterator trainingData)
Set the training dataFitConfig
train(@NonNull MultiDataSetIterator trainingData, int epochs)
Set the training data and number of epochsFitConfig
validate(DataSetIterator validationData)
Set the validation dataFitConfig
validate(DataSetIterator validationData, int validationFrequency)
Set the validation data and frequencyFitConfig
validate(MultiDataSetIterator validationData)
Set the validation dataFitConfig
validate(MultiDataSetIterator validationData, int validationFrequency)
Set the validation data and frequencyFitConfig
validationFrequency(int validationFrequency)
Set the validation frequency.
-
-
-
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 frequencySpecifically, validation will be preformed when i % validationFrequency == 0
-
validate
public FitConfig validate(DataSetIterator validationData, int validationFrequency)
Set the validation data and frequencySpecifically, validation will be preformed when i % validationFrequency == 0
-
listeners
public FitConfig listeners(@NonNull @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 theTrainingConfig
, loss values, and timing information).
-
-