Class EarlyStoppingGraphTrainer
- java.lang.Object
-
- org.deeplearning4j.earlystopping.trainer.BaseEarlyStoppingTrainer<ComputationGraph>
-
- org.deeplearning4j.earlystopping.trainer.EarlyStoppingGraphTrainer
-
- All Implemented Interfaces:
IEarlyStoppingTrainer<ComputationGraph>
public class EarlyStoppingGraphTrainer extends BaseEarlyStoppingTrainer<ComputationGraph>
-
-
Field Summary
-
Fields inherited from class org.deeplearning4j.earlystopping.trainer.BaseEarlyStoppingTrainer
esConfig, model
-
-
Constructor Summary
Constructors Constructor Description EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, DataSetIterator train)EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, DataSetIterator train, EarlyStoppingListener<ComputationGraph> listener)Constructor for training using aDataSetIteratorEarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, MultiDataSetIterator train, EarlyStoppingListener<ComputationGraph> listener)Constructor for training using aMultiDataSetIterator
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description protected voidfit(MultiDataSet mds)protected voidfit(DataSet ds)protected voidpretrain(MultiDataSet mds)protected voidpretrain(DataSet ds)-
Methods inherited from class org.deeplearning4j.earlystopping.trainer.BaseEarlyStoppingTrainer
fit, fit, pretrain, reset, setListener, triggerEpochListeners
-
-
-
-
Constructor Detail
-
EarlyStoppingGraphTrainer
public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, DataSetIterator train)
- Parameters:
esConfig- Configurationnet- Network to train using early stoppingtrain- DataSetIterator for training the network
-
EarlyStoppingGraphTrainer
public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, DataSetIterator train, EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using aDataSetIterator- Parameters:
esConfig- Configurationnet- Network to train using early stoppingtrain- DataSetIterator for training the networklistener- Early stopping listener. May be null.
-
EarlyStoppingGraphTrainer
public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, MultiDataSetIterator train, EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using aMultiDataSetIterator- Parameters:
esConfig- Configurationnet- Network to train using early stoppingtrain- DataSetIterator for training the networklistener- Early stopping listener. May be null.
-
-
Method Detail
-
fit
protected void fit(DataSet ds)
- Specified by:
fitin classBaseEarlyStoppingTrainer<ComputationGraph>
-
fit
protected void fit(MultiDataSet mds)
- Specified by:
fitin classBaseEarlyStoppingTrainer<ComputationGraph>
-
pretrain
protected void pretrain(DataSet ds)
- Specified by:
pretrainin classBaseEarlyStoppingTrainer<ComputationGraph>
-
pretrain
protected void pretrain(MultiDataSet mds)
- Specified by:
pretrainin classBaseEarlyStoppingTrainer<ComputationGraph>
-
-