org.deeplearning4j.earlystopping.trainer

Class EarlyStoppingTrainer

java.lang.Object

org.deeplearning4j.earlystopping.trainer.BaseEarlyStoppingTrainer<MultiLayerNetwork>

org.deeplearning4j.earlystopping.trainer.EarlyStoppingTrainer

All Implemented Interfaces:

IEarlyStoppingTrainer<MultiLayerNetwork>

public class EarlyStoppingTrainer
extends BaseEarlyStoppingTrainer<MultiLayerNetwork>

Field Summary

Fields inherited from class org.deeplearning4j.earlystopping.trainer.BaseEarlyStoppingTrainer

esConfig, model

Constructor Summary

Constructors

Constructor and Description
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> earlyStoppingConfiguration, MultiLayerConfiguration configuration, DataSetIterator train)
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, DataSetIterator train)
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, DataSetIterator train, EarlyStoppingListener<MultiLayerNetwork> listener)

Method Summary

Modifier and Type	Method and Description
protected void	fit(DataSet ds)
protected void	fit(MultiDataSet mds)
protected void	pretrain(DataSet ds)
protected void	pretrain(MultiDataSet mds)

Methods inherited from class org.deeplearning4j.earlystopping.trainer.BaseEarlyStoppingTrainer

fit, fit, pretrain, reset, setListener, triggerEpochListeners

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

EarlyStoppingTrainer

public EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> earlyStoppingConfiguration,
                            MultiLayerConfiguration configuration,
                            DataSetIterator train)

EarlyStoppingTrainer

public EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
                            MultiLayerNetwork net,
                            DataSetIterator train)

EarlyStoppingTrainer

public EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
                            MultiLayerNetwork net,
                            DataSetIterator train,
                            EarlyStoppingListener<MultiLayerNetwork> listener)

Method Detail

fit

protected void fit(DataSet ds)

Specified by:

fit in class BaseEarlyStoppingTrainer<MultiLayerNetwork>

fit

protected void fit(MultiDataSet mds)

Specified by:

fit in class BaseEarlyStoppingTrainer<MultiLayerNetwork>

pretrain

protected void pretrain(DataSet ds)

Specified by:

pretrain in class BaseEarlyStoppingTrainer<MultiLayerNetwork>

pretrain

protected void pretrain(MultiDataSet mds)

Specified by:

pretrain in class BaseEarlyStoppingTrainer<MultiLayerNetwork>