public abstract class BaseEarlyStoppingTrainer<T extends Model> extends java.lang.Object implements IEarlyStoppingTrainer<T>
MultiLayerNetwork or a ComputationGraph via early stopping| Modifier 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()
IEarlyStoppingTrainerfit in interface IEarlyStoppingTrainer<T extends Model>public void setListener(EarlyStoppingListener<T> listener)
IEarlyStoppingTrainersetListener in interface IEarlyStoppingTrainer<T extends Model>protected void triggerEpochListeners(boolean epochStart,
Model model,
int epochNum)
protected void reset()