org.deeplearning4j.optimize.listeners

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 and Description
protected EvaluationCallback	callback This callback will be invoked after evaluation finished
protected 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 and Description
EvaluativeListener(@NonNull DataSet dataSet, int frequency, @NonNull InvocationType type)
EvaluativeListener(@NonNull DataSet dataSet, int frequency, @NonNull InvocationType type, IEvaluation... evaluations)
EvaluativeListener(@NonNull DataSetIterator iterator, int frequency) Evaluation will be launched after each *frequency* iterations, with Evaluation datatype
EvaluativeListener(@NonNull DataSetIterator iterator, int frequency, IEvaluation... evaluations) Evaluation will be launched after each *frequency* iteration
EvaluativeListener(@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* iteration
EvaluativeListener(@NonNull MultiDataSet multiDataSet, int frequency, @NonNull InvocationType type)
EvaluativeListener(@NonNull MultiDataSet multiDataSet, int frequency, @NonNull InvocationType type, IEvaluation... evaluations)
EvaluativeListener(@NonNull MultiDataSetIterator iterator, int frequency) Evaluation will be launched after each *frequency* iterations, with Evaluation datatype
EvaluativeListener(@NonNull MultiDataSetIterator iterator, int frequency, IEvaluation... evaluations) Evaluation will be launched after each *frequency* iteration
EvaluativeListener(@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* iteration

Method Summary

Modifier and Type	Method and 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 iteration
void	onEpochEnd(Model model) Called once at the end of each epoch, when using methods such as MultiLayerNetwork.fit(DataSetIterator), ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator)
void	onEpochStart(Model model) Called once at the start of each epoch, when using methods such as MultiLayerNetwork.fit(DataSetIterator), ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator)

Methods inherited from class org.deeplearning4j.optimize.api.BaseTrainingListener

onBackwardPass, onForwardPass, onForwardPass, onGradientCalculation

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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, with Evaluation datatype

Parameters:

iterator - Iterator to provide data for evaluation

frequency - 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 evaluation

frequency - Frequency (in number of iterations/epochs according to the invocation type) to perform evaluation

type - 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, with Evaluation datatype

Parameters:

iterator - Iterator to provide data for evaluation

frequency - 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 evaluation

frequency - Frequency (in number of iterations/epochs according to the invocation type) to perform evaluation

type - 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 evaluation

frequency - Frequency (in number of iterations) to perform evaluation

evaluations - 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 evaluation

frequency - Frequency (in number of iterations/epochs according to the invocation type) to perform evaluation

type - Type of value for 'frequency' - iteration end, epoch end, etc

evaluations - 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 evaluation

frequency - Frequency (in number of iterations) to perform evaluation

evaluations - 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 evaluation

frequency - Frequency (in number of iterations/epochs according to the invocation type) to perform evaluation

type - Type of value for 'frequency' - iteration end, epoch end, etc

evaluations - 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 interface TrainingListener

Overrides:

iterationDone in class BaseTrainingListener

Parameters:

model - the model iterating

iteration - 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 as MultiLayerNetwork.fit(DataSetIterator), ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator)

Specified by:

onEpochStart in interface TrainingListener

Overrides:

onEpochStart in class BaseTrainingListener

onEpochEnd

public void onEpochEnd(Model model)

Description copied from interface: TrainingListener

Called once at the end of each epoch, when using methods such as MultiLayerNetwork.fit(DataSetIterator), ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator)

Specified by:

onEpochEnd in interface TrainingListener

Overrides:

onEpochEnd in class BaseTrainingListener

invokeListener

protected void invokeListener(Model model)

evalAtIndex

protected void evalAtIndex(IEvaluation evaluation,
                           INDArray[] labels,
                           INDArray[] predictions,
                           int index)