org.deeplearning4j.optimize.listeners

Class CollectScoresListener

java.lang.Object

org.deeplearning4j.optimize.api.BaseTrainingListener

org.deeplearning4j.optimize.listeners.CollectScoresListener

All Implemented Interfaces:

Serializable, TrainingListener

public class CollectScoresListener
extends BaseTrainingListener
implements Serializable

A simple listener that collects scores to a list every N iterations. Can also optionally log the score.

Author:

Alex Black

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
CollectScoresListener(int frequency)
CollectScoresListener(int frequency, boolean logScore)

Method Summary

Modifier and Type	Method and Description
void	iterationDone(Model model, int iteration, int epoch) Event listener for each iteration.

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

onBackwardPass, onEpochEnd, onEpochStart, onForwardPass, onForwardPass, onGradientCalculation

Methods inherited from class java.lang.Object

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

Constructor Detail

CollectScoresListener

public CollectScoresListener(int frequency)

CollectScoresListener

public CollectScoresListener(int frequency,
                             boolean logScore)

Method Detail

iterationDone

public void iterationDone(Model model,
                          int iteration,
                          int epoch)

Description copied from interface: TrainingListener

Event listener for each iteration. Called once, after each parameter update has ocurred while training the network

Specified by:

iterationDone in interface TrainingListener

Overrides:

iterationDone in class BaseTrainingListener

Parameters:

model - the model iterating

iteration - the iteration