org.deeplearning4j.earlystopping.termination

Class ScoreImprovementEpochTerminationCondition

java.lang.Object

org.deeplearning4j.earlystopping.termination.ScoreImprovementEpochTerminationCondition

All Implemented Interfaces:

Serializable, EpochTerminationCondition

public class ScoreImprovementEpochTerminationCondition
extends Object
implements EpochTerminationCondition

Terminate training if best model score does not improve for N epochs

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ScoreImprovementEpochTerminationCondition(int maxEpochsWithNoImprovement)
ScoreImprovementEpochTerminationCondition(int maxEpochsWithNoImprovement, double minImprovement)

Method Summary

Modifier and Type	Method and Description
void	initialize() Initialize the epoch termination condition (often a no-op)
boolean	terminate(int epochNum, double score, boolean minimize) Should the early stopping training terminate at this epoch, based on the calculated score and the epoch number? Returns true if training should terminated, or false otherwise
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

ScoreImprovementEpochTerminationCondition

public ScoreImprovementEpochTerminationCondition(int maxEpochsWithNoImprovement)

ScoreImprovementEpochTerminationCondition

public ScoreImprovementEpochTerminationCondition(int maxEpochsWithNoImprovement,
                                                 double minImprovement)

Method Detail

initialize

public void initialize()

Description copied from interface: EpochTerminationCondition

Initialize the epoch termination condition (often a no-op)

Specified by:

initialize in interface EpochTerminationCondition

terminate

public boolean terminate(int epochNum,
                         double score,
                         boolean minimize)

Description copied from interface: EpochTerminationCondition

Should the early stopping training terminate at this epoch, based on the calculated score and the epoch number? Returns true if training should terminated, or false otherwise

Specified by:

terminate in interface EpochTerminationCondition

Parameters:

epochNum - Number of the last completed epoch (starting at 0)

score - Score calculate for this epoch

Returns:

Whether training should be terminated at this epoch

toString

public String toString()

Overrides:

toString in class Object