Module training 

Training utilities for neural networks using the array protocol.

This module provides utilities for training neural networks using the array protocol, including datasets, dataloaders, loss functions, and training loops.

Structs

CrossEntropyLoss

Cross-entropy loss.

DataLoader

Data loader for batching and shuffling datasets.

InMemoryDataset

In-memory dataset with arrays.

MSELoss

Mean squared error loss.

Metrics

Metrics for evaluating model performance.

ProgressCallback

Progress bar callback for displaying training progress.

Trainer

Model trainer for neural networks.

Traits

Dataset

Dataset trait for providing data samples.

Loss

Loss function trait.

TrainingCallback

Training progress callback trait.

Type Aliases

BatchData

Type alias for batch data