EarlyStopping

class EarlyStopping(monitor: KProperty1<EpochTrainingEvent, Double?>, minDelta: Double, patience: Int, verbose: Boolean, mode: EarlyStoppingMode, baseline: Double, restoreBestWeights: Boolean) : Callback

This callback stops training when a monitored metric has stopped improving.

Assuming the goal of a training is to minimize the loss. With this, the metric to be monitored would be 'loss', and mode would be 'min'. A model.fit() training loop will check at end of every epoch whether the loss is no longer decreasing, considering the min_delta and patience if applicable. Once it's found no longer decreasing, model.stop_training is marked True and the training terminates.

The quantity to be monitored needs to be available in logs. To make it so, pass the loss or metrics at model.compile().

Constructors

EarlyStopping

fun EarlyStopping(monitor: KProperty1<EpochTrainingEvent, Double?> = EpochTrainingEvent::lossValue, minDelta: Double = 0.0, patience: Int = 0, verbose: Boolean = false, mode: EarlyStoppingMode = EarlyStoppingMode.AUTO, baseline: Double = 0.001, restoreBestWeights: Boolean = false)

Creates an EarlyStopping Callback.

Functions

onEpochBegin

open fun onEpochBegin(epoch: Int, logs: TrainingHistory)

Called at the start of an epoch during training phase.

onEpochEnd

open override fun onEpochEnd(epoch: Int, event: EpochTrainingEvent, logs: TrainingHistory)

Called at the end of an epoch during training phase.

onPredictBatchBegin

open fun onPredictBatchBegin(batch: Int, batchSize: Int)

Called at the beginning of a batch during prediction phase.

onPredictBatchEnd

open fun onPredictBatchEnd(batch: Int, batchSize: Int)

Called at the end of a batch during prediction phase.

onPredictBegin

open fun onPredictBegin()

Called at the beginning of prediction.

onPredictEnd

open fun onPredictEnd()

Called at the end of prediction.

onTestBatchBegin

open fun onTestBatchBegin(batch: Int, batchSize: Int, logs: History)

Called at the beginning of a batch during evaluation phase. Also called at the beginning of a validation batch during validation phase, if validation data is provided.

onTestBatchEnd

open fun onTestBatchEnd(batch: Int, batchSize: Int, event: BatchEvent?, logs: History)

Called at the end of a batch during evaluation phase. Also called at the end of a validation batch during validation phase, if validation data is provided.

onTestBegin

open fun onTestBegin()

Called at the beginning of evaluation or validation.

onTestEnd

open fun onTestEnd(logs: History)

Called at the end of evaluation or validation.

onTrainBatchBegin

open fun onTrainBatchBegin(batch: Int, batchSize: Int, logs: TrainingHistory)

Called at the beginning of a batch during training phase.

onTrainBatchEnd

open fun onTrainBatchEnd(batch: Int, batchSize: Int, event: BatchTrainingEvent, logs: TrainingHistory)

Called at the end of a batch during training phase.

onTrainBegin

open override fun onTrainBegin()

Called at the beginning of training.

onTrainEnd

open override fun onTrainEnd(logs: TrainingHistory)

Called at the end of training. This method is empty. Extend this class to handle this event.