fit
fun fit(epochs: Int = 1, batchSize: Int = 32, lossCallback: (List<Float>) -> Unit? = null): List<Double>
Fit the local model using trainingSamples for epochs epochs with batch size batchSize.
Thread-safe, and block operations on trainingSamples.
Return
List of average training losses for each epoch.
Parameters
lossCallback
Called after every epoch with the List of training losses.