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.