public abstract class BaseEarlyStoppingTrainer<T extends Model> extends java.lang.Object implements IEarlyStoppingTrainer<T>
MultiLayerNetwork
or a ComputationGraph
via early stoppingModifier and Type | Field and Description |
---|---|
protected EarlyStoppingConfiguration<T> |
esConfig |
protected T |
model |
Modifier | Constructor and Description |
---|---|
protected |
BaseEarlyStoppingTrainer(EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
DataSetIterator train,
MultiDataSetIterator trainMulti,
EarlyStoppingListener<T> listener) |
Modifier and Type | Method and Description |
---|---|
EarlyStoppingResult<T> |
fit()
Conduct early stopping training
|
protected abstract void |
fit(DataSet ds) |
protected abstract void |
fit(MultiDataSet mds) |
protected void |
reset() |
void |
setListener(EarlyStoppingListener<T> listener)
Set the early stopping listener
|
protected void |
triggerEpochListeners(boolean epochStart,
Model model,
int epochNum) |
protected final EarlyStoppingConfiguration<T extends Model> esConfig
protected BaseEarlyStoppingTrainer(EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, DataSetIterator train, MultiDataSetIterator trainMulti, EarlyStoppingListener<T> listener)
protected abstract void fit(DataSet ds)
protected abstract void fit(MultiDataSet mds)
public EarlyStoppingResult<T> fit()
IEarlyStoppingTrainer
fit
in interface IEarlyStoppingTrainer<T extends Model>
public void setListener(EarlyStoppingListener<T> listener)
IEarlyStoppingTrainer
setListener
in interface IEarlyStoppingTrainer<T extends Model>
protected void triggerEpochListeners(boolean epochStart, Model model, int epochNum)
protected void reset()