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

Types

Callback
Link copied to clipboard
open class Callback
Content copied to clipboard

Base class used to build new callbacks.

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

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

EarlyStoppingMode
Link copied to clipboard
enum EarlyStoppingMode : Enum<EarlyStoppingMode>
Content copied to clipboard

This enum describes a few strategies of training stopping.

TerminateOnNaN
Link copied to clipboard
class TerminateOnNaN : Callback
Content copied to clipboard

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

TimeStopping
Link copied to clipboard
class TimeStopping(seconds: Int, verbose: Boolean) : Callback
Content copied to clipboard

Callback for stopping training of a model when a specified amount of time has passed.