org.deeplearning4j.earlystopping.trainer

Class BaseEarlyStoppingTrainer<T extends Model>

java.lang.Object

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

All Implemented Interfaces:

IEarlyStoppingTrainer<T>

Direct Known Subclasses:

EarlyStoppingGraphTrainer, EarlyStoppingTrainer

public abstract class BaseEarlyStoppingTrainer<T extends Model>
extends java.lang.Object
implements IEarlyStoppingTrainer<T>

Base/abstract class for conducting early stopping training locally (single machine).
Can be used to train a MultiLayerNetwork or a ComputationGraph via early stopping

Field Summary

Fields

Modifier and Type	Field and Description
protected EarlyStoppingConfiguration<T>	esConfig
protected T	model

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	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)

Method Summary

Modifier and Type	Method and Description
EarlyStoppingResult<T>	fit() Conduct early stopping training
protected abstract void	fit(org.nd4j.linalg.dataset.DataSet ds)
protected abstract void	fit(org.nd4j.linalg.dataset.api.MultiDataSet mds)
protected void	reset()
void	setListener(EarlyStoppingListener<T> listener) Set the early stopping listener

Methods inherited from class java.lang.Object

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

Field Detail

model

protected T extends Model model

esConfig

protected final EarlyStoppingConfiguration<T extends Model> esConfig

Constructor Detail

BaseEarlyStoppingTrainer

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

Method Detail

fit

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

fit

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

fit

public EarlyStoppingResult<T> fit()

Description copied from interface: IEarlyStoppingTrainer

Conduct early stopping training

Specified by:

fit in interface IEarlyStoppingTrainer<T extends Model>

setListener

public void setListener(EarlyStoppingListener<T> listener)

Description copied from interface: IEarlyStoppingTrainer

Set the early stopping listener

Specified by:

setListener in interface IEarlyStoppingTrainer<T extends Model>

reset

protected void reset()