Package org.jetbrains.kotlinx.dl.api.core.callback

Types

Callback

Base class used to build new callbacks.

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.

EarlyStoppingMode

This enum describes a few strategies of training stopping.

TerminateOnNaN

This callback is used to stop the training if loss is not a number (NaN or INFINITY).