weka.dl4j.listener

Class EpochListener

java.lang.Object

org.deeplearning4j.optimize.api.BaseTrainingListener

weka.dl4j.listener.TrainingListener

weka.dl4j.listener.EpochListener

All Implemented Interfaces:

java.io.Serializable, org.deeplearning4j.optimize.api.TrainingListener, weka.core.OptionHandler

public class EpochListener
extends TrainingListener

A listener that prints the model score every epoch. Inspired by ScoreIterationListener written by Adam Gibson.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
EpochListener()

Method Summary

Modifier and Type	Method and Description
int	getN()
java.lang.String	globalInfo() Returns a string describing this search method
boolean	isIntermediateEvaluationsEnabled()
void	log(java.lang.String msg) Log a message
void	onEpochEnd(org.deeplearning4j.nn.api.Model model)
void	setIntermediateEvaluationsEnabled(boolean intermediateEvaluationsEnabled)
void	setN(int evaluateEveryNEpochs)

Methods inherited from class weka.dl4j.listener.TrainingListener

getOptions, init, iterationDone, listOptions, setOptions

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

onBackwardPass, onEpochStart, onForwardPass, onForwardPass, onGradientCalculation

Methods inherited from class java.lang.Object

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

Methods inherited from interface weka.core.OptionHandler

makeCopy

Constructor Detail

EpochListener

public EpochListener()

Method Detail

onEpochEnd

public void onEpochEnd(org.deeplearning4j.nn.api.Model model)

Specified by:

onEpochEnd in interface org.deeplearning4j.optimize.api.TrainingListener

Overrides:

onEpochEnd in class org.deeplearning4j.optimize.api.BaseTrainingListener

log

public void log(java.lang.String msg)

Description copied from class: TrainingListener

Log a message

Specified by:

log in class TrainingListener

Parameters:

msg - Message

getN

public int getN()

setN

@OptionMetadata(displayName="evaluate every N epochs",
                description="Evaluate every N epochs (default = 5).",
                commandLineParamName="n",
                commandLineParamSynopsis="-n <int>",
                displayOrder=0)
public void setN(int evaluateEveryNEpochs)

isIntermediateEvaluationsEnabled

@OptionMetadata(displayName="enable intermediate evaluations",
                description="Enable intermediate evaluations (default = true).",
                commandLineParamName="eval",
                commandLineParamSynopsis="-eval <boolean>",
                displayOrder=0)
public boolean isIntermediateEvaluationsEnabled()

setIntermediateEvaluationsEnabled

public void setIntermediateEvaluationsEnabled(boolean intermediateEvaluationsEnabled)

globalInfo

public java.lang.String globalInfo()

Returns a string describing this search method

Returns:

a description of the search method suitable for displaying in the explorer/experimenter gui