feature-training

provider(?)

feature-training/org.pointyware.disco.training.viewmodels/TrainingViewModel

TrainingViewModel

class TrainingViewModel(controller: TrainingController) : ViewModel

This view model maintains the ui state for training a neural network.

Constructors

TrainingViewModel

constructor(controller: TrainingController)

Properties

val state: StateFlow<TrainingUiState>

The current state of the training UI.

Functions

expect open fun addCloseable(closeable: AutoCloseable)

expect fun addCloseable(key: String, closeable: AutoCloseable)

expect fun <T : AutoCloseable> getCloseable(key: String): T?

fun loadNetwork(networkId: String)

Load a neural network for training.

Reset current training session.

fun onRun(epochs: Int)

fun onStop()