| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff | |
| org.nd4j.autodiff.samediff.config |
| Modifier and Type | Method and Description |
|---|---|
FitConfig |
SameDiff.fit()
Set up for a fit operation using
FitConfig. |
| Modifier and Type | Method and Description |
|---|---|
FitConfig |
FitConfig.epochs(int epochs)
Set the number of epochs to train for
|
FitConfig |
FitConfig.listeners(Listener... listeners)
Add listeners for this operation
|
FitConfig |
FitConfig.train(@NonNull DataSetIterator trainingData)
Set the training data
|
FitConfig |
FitConfig.train(@NonNull DataSetIterator trainingData,
int epochs)
Set the training data and number of epochs
|
FitConfig |
FitConfig.train(@NonNull MultiDataSetIterator trainingData)
Set the training data
|
FitConfig |
FitConfig.train(@NonNull MultiDataSetIterator trainingData,
int epochs)
Set the training data and number of epochs
|
FitConfig |
FitConfig.validate(DataSetIterator validationData)
Set the validation data
|
FitConfig |
FitConfig.validate(DataSetIterator validationData,
int validationFrequency)
Set the validation data and frequency
|
FitConfig |
FitConfig.validate(MultiDataSetIterator validationData)
Set the validation data
|
FitConfig |
FitConfig.validate(MultiDataSetIterator validationData,
int validationFrequency)
Set the validation data and frequency
|
FitConfig |
FitConfig.validationFrequency(int validationFrequency)
Set the validation frequency.
|
Copyright © 2021. All rights reserved.