Class EvaluativeListener
- java.lang.Object
-
- org.deeplearning4j.optimize.api.BaseTrainingListener
-
- org.deeplearning4j.optimize.listeners.EvaluativeListener
-
- All Implemented Interfaces:
TrainingListener
public class EvaluativeListener extends BaseTrainingListener
-
-
Field Summary
Fields Modifier and Type Field Description protected EvaluationCallback
callback
This callback will be invoked after evaluation finishedprotected DataSet
ds
protected DataSetIterator
dsIterator
protected IEvaluation[]
evaluations
protected int
frequency
protected AtomicLong
invocationCount
protected InvocationType
invocationType
protected ThreadLocal<AtomicLong>
iterationCount
protected MultiDataSet
mds
protected MultiDataSetIterator
mdsIterator
-
Constructor Summary
Constructors Constructor Description EvaluativeListener(@NonNull DataSetIterator iterator, int frequency)
Evaluation will be launched after each *frequency* iterations, withEvaluation
datatypeEvaluativeListener(@NonNull DataSetIterator iterator, int frequency, @NonNull InvocationType type)
EvaluativeListener(@NonNull DataSetIterator iterator, int frequency, @NonNull InvocationType type, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iterationEvaluativeListener(@NonNull DataSetIterator iterator, int frequency, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iterationEvaluativeListener(@NonNull MultiDataSetIterator iterator, int frequency)
Evaluation will be launched after each *frequency* iterations, withEvaluation
datatypeEvaluativeListener(@NonNull MultiDataSetIterator iterator, int frequency, @NonNull InvocationType type)
EvaluativeListener(@NonNull MultiDataSetIterator iterator, int frequency, @NonNull InvocationType type, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iterationEvaluativeListener(@NonNull MultiDataSetIterator iterator, int frequency, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iterationEvaluativeListener(@NonNull DataSet dataSet, int frequency, @NonNull InvocationType type)
EvaluativeListener(@NonNull DataSet dataSet, int frequency, @NonNull InvocationType type, IEvaluation... evaluations)
EvaluativeListener(@NonNull MultiDataSet multiDataSet, int frequency, @NonNull InvocationType type)
EvaluativeListener(@NonNull MultiDataSet multiDataSet, int frequency, @NonNull InvocationType type, IEvaluation... evaluations)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description protected void
evalAtIndex(IEvaluation evaluation, INDArray[] labels, INDArray[] predictions, int index)
protected void
invokeListener(Model model)
void
iterationDone(Model model, int iteration, int epoch)
Event listener for each iterationvoid
onEpochEnd(Model model)
Called once at the end of each epoch, when using methods such asMultiLayerNetwork.fit(DataSetIterator)
,ComputationGraph.fit(DataSetIterator)
orComputationGraph.fit(MultiDataSetIterator)
void
onEpochStart(Model model)
Called once at the start of each epoch, when using methods such asMultiLayerNetwork.fit(DataSetIterator)
,ComputationGraph.fit(DataSetIterator)
orComputationGraph.fit(MultiDataSetIterator)
-
Methods inherited from class org.deeplearning4j.optimize.api.BaseTrainingListener
onBackwardPass, onForwardPass, onForwardPass, onGradientCalculation
-
-
-
-
Field Detail
-
iterationCount
protected transient ThreadLocal<AtomicLong> iterationCount
-
frequency
protected int frequency
-
invocationCount
protected AtomicLong invocationCount
-
dsIterator
protected transient DataSetIterator dsIterator
-
mdsIterator
protected transient MultiDataSetIterator mdsIterator
-
ds
protected DataSet ds
-
mds
protected MultiDataSet mds
-
evaluations
protected IEvaluation[] evaluations
-
invocationType
protected InvocationType invocationType
-
callback
protected transient EvaluationCallback callback
This callback will be invoked after evaluation finished
-
-
Constructor Detail
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull DataSetIterator iterator, int frequency)
Evaluation will be launched after each *frequency* iterations, withEvaluation
datatype- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations) to perform evaluation
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull DataSetIterator iterator, int frequency, @NonNull @NonNull InvocationType type)
- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations/epochs according to the invocation type) to perform evaluationtype
- Type of value for 'frequency' - iteration end, epoch end, etc
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull MultiDataSetIterator iterator, int frequency)
Evaluation will be launched after each *frequency* iterations, withEvaluation
datatype- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations) to perform evaluation
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull MultiDataSetIterator iterator, int frequency, @NonNull @NonNull InvocationType type)
- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations/epochs according to the invocation type) to perform evaluationtype
- Type of value for 'frequency' - iteration end, epoch end, etc
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull DataSetIterator iterator, int frequency, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iteration- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations) to perform evaluationevaluations
- Type of evalutions to perform
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull DataSetIterator iterator, int frequency, @NonNull @NonNull InvocationType type, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iteration- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations/epochs according to the invocation type) to perform evaluationtype
- Type of value for 'frequency' - iteration end, epoch end, etcevaluations
- Type of evalutions to perform
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull MultiDataSetIterator iterator, int frequency, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iteration- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations) to perform evaluationevaluations
- Type of evalutions to perform
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull MultiDataSetIterator iterator, int frequency, @NonNull @NonNull InvocationType type, IEvaluation... evaluations)
Evaluation will be launched after each *frequency* iteration- Parameters:
iterator
- Iterator to provide data for evaluationfrequency
- Frequency (in number of iterations/epochs according to the invocation type) to perform evaluationtype
- Type of value for 'frequency' - iteration end, epoch end, etcevaluations
- Type of evalutions to perform
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull DataSet dataSet, int frequency, @NonNull @NonNull InvocationType type)
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull MultiDataSet multiDataSet, int frequency, @NonNull @NonNull InvocationType type)
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull DataSet dataSet, int frequency, @NonNull @NonNull InvocationType type, IEvaluation... evaluations)
-
EvaluativeListener
public EvaluativeListener(@NonNull @NonNull MultiDataSet multiDataSet, int frequency, @NonNull @NonNull InvocationType type, IEvaluation... evaluations)
-
-
Method Detail
-
iterationDone
public void iterationDone(Model model, int iteration, int epoch)
Event listener for each iteration- Specified by:
iterationDone
in interfaceTrainingListener
- Overrides:
iterationDone
in classBaseTrainingListener
- Parameters:
model
- the model iteratingiteration
- the iteration
-
onEpochStart
public void onEpochStart(Model model)
Description copied from interface:TrainingListener
Called once at the start of each epoch, when using methods such asMultiLayerNetwork.fit(DataSetIterator)
,ComputationGraph.fit(DataSetIterator)
orComputationGraph.fit(MultiDataSetIterator)
- Specified by:
onEpochStart
in interfaceTrainingListener
- Overrides:
onEpochStart
in classBaseTrainingListener
-
onEpochEnd
public void onEpochEnd(Model model)
Description copied from interface:TrainingListener
Called once at the end of each epoch, when using methods such asMultiLayerNetwork.fit(DataSetIterator)
,ComputationGraph.fit(DataSetIterator)
orComputationGraph.fit(MultiDataSetIterator)
- Specified by:
onEpochEnd
in interfaceTrainingListener
- Overrides:
onEpochEnd
in classBaseTrainingListener
-
invokeListener
protected void invokeListener(Model model)
-
evalAtIndex
protected void evalAtIndex(IEvaluation evaluation, INDArray[] labels, INDArray[] predictions, int index)
-
-