Package 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 Object
implements IEarlyStoppingTrainer<T>

Field Summary

Fields

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

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	BaseEarlyStoppingTrainer(EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, DataSetIterator train, MultiDataSetIterator trainMulti, EarlyStoppingListener<T> listener)

Method Summary

Modifier and Type	Method	Description
EarlyStoppingResult<T>	fit()	Conduct early stopping training
protected EarlyStoppingResult<T>	fit(boolean pretrain)
protected abstract void	fit(MultiDataSet mds)
protected abstract void	fit(DataSet ds)
EarlyStoppingResult<T>	pretrain()
protected abstract void	pretrain(MultiDataSet mds)
protected abstract void	pretrain(DataSet ds)
protected void	reset()
void	setListener(EarlyStoppingListener<T> listener)	Set the early stopping listener
protected void	triggerEpochListeners(boolean epochStart, Model model, int epochNum)

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,
                                   DataSetIterator train,
                                   MultiDataSetIterator trainMulti,
                                   EarlyStoppingListener<T> listener)

Method Detail

fit

protected abstract void fit(DataSet ds)

fit

protected abstract void fit(MultiDataSet mds)

pretrain

protected abstract void pretrain(DataSet ds)

pretrain

protected abstract void pretrain(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>

pretrain

public EarlyStoppingResult<T> pretrain()

Specified by:

pretrain in interface IEarlyStoppingTrainer<T extends Model>

fit

protected EarlyStoppingResult<T> fit(boolean pretrain)

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>

triggerEpochListeners

protected void triggerEpochListeners(boolean epochStart,
                                     Model model,
                                     int epochNum)

reset

protected void reset()