| Package | Description |
|---|---|
| org.deeplearning4j.earlystopping.trainer |
| Modifier and Type | Method and Description |
|---|---|
void |
IEarlyStoppingTrainer.setListener(EarlyStoppingListener<T> listener)
Set the early stopping listener
|
void |
BaseEarlyStoppingTrainer.setListener(EarlyStoppingListener<T> listener) |
| Constructor and Description |
|---|
BaseEarlyStoppingTrainer(EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti,
EarlyStoppingListener<T> listener) |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
DataSetIterator |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
MultiDataSetIterator |
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
Copyright © 2018. All rights reserved.