public class EarlyStoppingParallelTrainer<T extends org.deeplearning4j.nn.api.Model> extends Object implements org.deeplearning4j.earlystopping.trainer.IEarlyStoppingTrainer<T>
MultiLayerNetwork or a ComputationGraph via early stopping.| Modifier and Type | Field and Description |
|---|---|
protected org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> |
esConfig |
protected T |
model |
protected org.deeplearning4j.earlystopping.termination.IterationTerminationCondition |
terminationReason |
| Constructor and Description |
|---|
EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti,
org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener,
int workers,
int prefetchBuffer,
int averagingFrequency) |
EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti,
org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener,
int workers,
int prefetchBuffer,
int averagingFrequency,
boolean reportScoreAfterAveraging,
boolean useLegacyAveraging) |
EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti,
int workers,
int prefetchBuffer,
int averagingFrequency) |
| Modifier and Type | Method and Description |
|---|---|
org.deeplearning4j.earlystopping.EarlyStoppingResult<T> |
fit() |
boolean |
getTermination() |
void |
incrementIteration() |
protected void |
reset() |
void |
setLatestScore(double latestScore) |
void |
setListener(org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener) |
void |
setTermination(boolean terminate) |
protected void |
setTerminationReason(org.deeplearning4j.earlystopping.termination.IterationTerminationCondition terminationReason) |
protected T extends org.deeplearning4j.nn.api.Model model
protected final org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T extends org.deeplearning4j.nn.api.Model> esConfig
protected volatile org.deeplearning4j.earlystopping.termination.IterationTerminationCondition terminationReason
public EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti, int workers, int prefetchBuffer, int averagingFrequency)
public EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti, org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener, int workers, int prefetchBuffer, int averagingFrequency)
public EarlyStoppingParallelTrainer(org.deeplearning4j.earlystopping.EarlyStoppingConfiguration<T> earlyStoppingConfiguration, T model, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator trainMulti, org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener, int workers, int prefetchBuffer, int averagingFrequency, boolean reportScoreAfterAveraging, boolean useLegacyAveraging)
protected void setTerminationReason(org.deeplearning4j.earlystopping.termination.IterationTerminationCondition terminationReason)
public org.deeplearning4j.earlystopping.EarlyStoppingResult<T> fit()
fit in interface org.deeplearning4j.earlystopping.trainer.IEarlyStoppingTrainer<T extends org.deeplearning4j.nn.api.Model>public void setLatestScore(double latestScore)
public void incrementIteration()
public void setTermination(boolean terminate)
public boolean getTermination()
public void setListener(org.deeplearning4j.earlystopping.listener.EarlyStoppingListener<T> listener)
setListener in interface org.deeplearning4j.earlystopping.trainer.IEarlyStoppingTrainer<T extends org.deeplearning4j.nn.api.Model>protected void reset()
Copyright © 2017. All rights reserved.