// Iris: Fisher (1936), UCI, CC BY 4.0. See THIRD_PARTY_DATA.md.
// Trains once before serving; Ctrl+C stops the foreground API.
import aner.api
import aner.dataset
import aner.nn
import aner.ml

let split = dataset.iris().split(test: 0.2, seed: 2026)
let network = nn.sequential(seed: 42)
    .dense(inputs: 4, outputs: 8).tanh()
    .dense(inputs: 8, outputs: 3)
let model = ml.classifier(network, standardize: true)
    .fit(split.train(), epochs: 200, rate: 0.1)
let app = api.app("Iris inference").model("/predict", model)
print(app.openapi())
app.serve()