Shared Model Training Commands

Example-side command runners for built-in runnable models.

The trainer API provides Trainer.new, trainer.train, and trained prediction handles. This file owns repository command plumbing: parse example flags, check local files, run training, and print the standard summary.

def NN.Examples.Models.TrainCommand.modelUsage (exeName : String) (dataOptions trainingOptions : Array String) (extraSections : Array String := #[]) : String

Help text for model commands with caller-supplied data and training options.

def NN.Examples.Models.TrainCommand.optimizerUsage (exeName : String) (dataOptions : Array String) : String

Standard optimizer and logging flags accepted by most model commands.

def NN.Examples.Models.TrainCommand.regressionCsv {σ τ : Spec.Shape} (exeName : String) (args : List String) (defaultCsv defaultLogPath : System.FilePath) (defaultSteps : ℕ := 1) (defaultLearningRate : Float := 1e-3) (banner : TorchLean.Runtime.Config → String) (train : TorchLean.Runtime.Config → Support.CsvTrainFlags → IO (TorchLean.Trainer.Result σ τ)) : IO UInt32

CSV-backed regression command using the public trainer API.

def NN.Examples.Models.TrainCommand.classificationNpy (exeName : String) (args : List String) (parseFlags : List String → Except String (Support.NpyModelTrainFlags × List String)) (banner : TorchLean.Runtime.Config → String) (train : TorchLean.Runtime.Config → Support.NpyModelTrainFlags → IO TorchLean.Trainer.Report) : IO UInt32

NPY-backed classifier command using the public trainer API.

def NN.Examples.Models.TrainCommand.regressionNpy {σ τ : Spec.Shape} (exeName : String) (args : List String) (parseFlags : List String → Except String (Support.NpyModelTrainFlags × List String)) (banner : TorchLean.Runtime.Config → String) (train : TorchLean.Runtime.Config → Support.NpyModelTrainFlags → IO (TorchLean.Trainer.Result σ τ)) : IO UInt32

NPY-backed regression command using the public trainer API.

def NN.Examples.Models.TrainCommand.forecastWindow {σ τ : Spec.Shape} (exeName : String) (args : List String) (parseFlags : List String → Except String (Support.ForecastWindowModelTrainFlags × List String)) (banner : TorchLean.Runtime.Config → String) (train : TorchLean.Runtime.Config → Support.ForecastWindowModelTrainFlags → IO (TorchLean.Trainer.Result σ τ)) : IO UInt32

Forecast-window regression command using the public trainer API.