Callback

Base class used to build new callbacks.

NOTE: This class contains empty methods, inherit it and override if you need functionality.

Callback methods are called during training, evaluation, prediction phases on each batch, epoch, start and end of the specific phase.

Constructors

Callback

fun Callback()

Functions

onEpochBegin

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

Called at the start of an epoch during training phase.

onEpochEnd

open 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