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>

Class for conducting early stopping training locally (single machine), for training a MultiLayerNetwork. To train a ComputationGraph, use EarlyStoppingGraphTrainer

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, 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)

Method Summary

Modifier and Type	Method and Description
protected void	fit(org.nd4j.linalg.dataset.DataSet ds)
protected void	fit(org.nd4j.linalg.dataset.api.MultiDataSet mds)

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

fit, reset, setListener

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,
                            org.nd4j.linalg.dataset.api.iterator.DataSetIterator train)

EarlyStoppingTrainer

public EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
                            MultiLayerNetwork net,
                            org.nd4j.linalg.dataset.api.iterator.DataSetIterator train)

EarlyStoppingTrainer

public EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
                            MultiLayerNetwork net,
                            org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
                            EarlyStoppingListener<MultiLayerNetwork> listener)

Method Detail

fit

protected void fit(org.nd4j.linalg.dataset.DataSet ds)

Specified by:

fit in class BaseEarlyStoppingTrainer<MultiLayerNetwork>

fit

protected void fit(org.nd4j.linalg.dataset.api.MultiDataSet mds)

Specified by:

fit in class BaseEarlyStoppingTrainer<MultiLayerNetwork>