org.deeplearning4j.earlystopping.trainer

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>

Class for conducting early stopping training locally (single machine).
Can be used to train a ComputationGraph

Field Summary

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

esConfig, model

Constructor Summary

Constructors

Constructor and Description
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, DataSetIterator train)
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, 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

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

EarlyStoppingGraphTrainer

public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
                                 ComputationGraph net,
                                 DataSetIterator train)

Parameters:

esConfig - Configuration

net - Network to train using early stopping

train - DataSetIterator for training the network

EarlyStoppingGraphTrainer

public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
                                 ComputationGraph net,
                                 DataSetIterator train,
                                 EarlyStoppingListener<ComputationGraph> listener)

Constructor for training using a DataSetIterator

Parameters:

esConfig - Configuration

net - Network to train using early stopping

train - DataSetIterator for training the network

listener - Early stopping listener. May be null.

EarlyStoppingGraphTrainer

public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
                                 ComputationGraph net,
                                 org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator train,
                                 EarlyStoppingListener<ComputationGraph> listener)

Constructor for training using a MultiDataSetIterator

Parameters:

esConfig - Configuration

net - Network to train using early stopping

train - DataSetIterator for training the network

listener - Early stopping listener. May be null.

Method Detail

fit

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

Specified by:

fit in class BaseEarlyStoppingTrainer<ComputationGraph>

fit

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

Specified by:

fit in class BaseEarlyStoppingTrainer<ComputationGraph>