public class FitConfig extends Object
SameDiff
training operation.
Used in SameDiff.fit()
.
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(DataSetIterator trainingData)
Set the training data
|
FitConfig |
train(DataSetIterator trainingData,
int epochs)
Set the training data and number of epochs
|
FitConfig |
train(MultiDataSetIterator trainingData)
Set the training data
|
FitConfig |
train(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.
|
public FitConfig(@NonNull SameDiff sd)
public FitConfig epochs(int epochs)
public FitConfig train(@NonNull MultiDataSetIterator trainingData)
public FitConfig train(@NonNull DataSetIterator trainingData)
public FitConfig train(@NonNull MultiDataSetIterator trainingData, int epochs)
public FitConfig train(@NonNull DataSetIterator trainingData, int epochs)
public FitConfig validate(MultiDataSetIterator validationData)
public FitConfig validate(DataSetIterator validationData)
public FitConfig validationFrequency(int validationFrequency)
Specifically, validation will be preformed when i % validationFrequency == 0
public FitConfig validate(MultiDataSetIterator validationData, int validationFrequency)
Specifically, validation will be preformed when i % validationFrequency == 0
public FitConfig validate(DataSetIterator validationData, int validationFrequency)
Specifically, validation will be preformed when i % validationFrequency == 0
public FitConfig listeners(@NonNull Listener... listeners)
public History exec()
History
object containing the history information for this training operation
(evaluations specified in the TrainingConfig
, loss values, and timing information).Copyright © 2019. All rights reserved.