Model Training

ML Training

Training loops combine data loading, forward pass, loss computation, and optimizer updates.

import ml
import datasets

var data = datasets::load_csv("examples/iris_dummy.csv")
var model = ml::Sequential([ml::Dense(4, 8), ml::ReLU(), ml::Dense(8, 3)])
var optimizer = ml::Adam(0.001)

for epoch from 0~20 {
    var loss = model.train_step(data::features, data::labels, optimizer)
    if epoch % 5 == 0 {
        display "epoch=" + epoch + " loss=" + loss
    }
}