Expand description
This crate works to augment the training process by providing datasets and loaders for common data formats.
Modules§
- dataset
- error
- The error module for external datasets and training;
- prelude
- trainer
- traits
- Additional traits and interfaces for working with datasets and data loaders.
Structs§
- Dataset
Base - A dataset is a collection of records and targets along with various other attributes useful for machine learning tasks
- Trainer
- The
Traineris a generalized model trainer that works to provide a common interface for training models over datasets.
Enums§
- Training
Error - The
TrainingErrortype enumerates the various errors that can occur during the training process.
Traits§
- AsDataset
- The
AsDatasettrait defines the conversion from some reference into a dataset. - Into
Dataset - Thge
IntoDatasettrait defines a method for consuming the caller to convert it into a dataset. - Model
Trainer - Records
- This trait generically defines the basic type of dataset that can be used throughout the framework.
Type Aliases§
- Training
Result - a type alias for a
Resultwith an error type ofTrainingError.