TrainingController

A Training Controller provides an interface to manage training of a model and exposes properties to observe the training state.

Properties

abstract val state: StateFlow<TrainingState>

abstract fun reset()

Stops the training process if it is currently running, and resets the training state.

abstract fun setEpochs(epochs: Int)

Sets the number of epochs to train the model. 0 means train indefinitely.

abstract fun start()

Starts the training process.

abstract fun stop()

Stop any ongoing training. The current epoch will be completed before stopping.

abstract suspend fun test(case: Exercise): Double