| Package | Description |
|---|---|
| org.deeplearning4j.earlystopping | |
| org.deeplearning4j.earlystopping.listener | |
| org.deeplearning4j.earlystopping.trainer |
| Modifier and Type | Method and Description |
|---|---|
EarlyStoppingConfiguration<T> |
EarlyStoppingConfiguration.Builder.build()
Create the early stopping configuration
|
| Modifier and Type | Method and Description |
|---|---|
void |
EarlyStoppingListener.onEpoch(int epochNum,
double score,
EarlyStoppingConfiguration<T> esConfig,
T net)
Method that is called at the end of each epoch completed during early stopping training
|
void |
EarlyStoppingListener.onStart(EarlyStoppingConfiguration<T> esConfig,
T net)
Method to be called when early stopping training is first started
|
| Modifier and Type | Field and Description |
|---|---|
protected EarlyStoppingConfiguration<T> |
BaseEarlyStoppingTrainer.esConfig |
| 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) |
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> earlyStoppingConfiguration,
MultiLayerConfiguration configuration,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
Copyright © 2018. All rights reserved.