Skip to main content

Crate dataset_ml

Crate dataset_ml 

Source
Expand description

Built-in dataset implementations for machine learning.

dataset-ml provides ready-to-use loaders for classic ML datasets built on top of dataset_core::Dataset. Each module is a worked example showing how to wrap Dataset<T, E> for a concrete data source: downloading from a URL, verifying a SHA-256 hash, parsing CSV records, and exposing typed accessors backed by ndarray.

§Datasets

ModuleSamplesFeaturesTask Type
iris1504Classification
breast_cancer56930Classification
boston_housing50613Regression
california_housing20,6408Regression
diabetes7688Classification
titanic89111Classification
palmer_penguins3447Classification
wine_recognition17813Classification
wine_quality::red_wine_quality1,59911Regression
wine_quality::white_wine_quality4,89811Regression

§Example

use dataset_ml::iris::Iris;

let iris = Iris::new("./data");
let (features, labels) = iris.data().unwrap();
assert_eq!(features.shape(), &[150, 4]);

All loaders are lazy: the first call downloads and parses the file, every subsequent call returns a cached reference. See the individual module docs for features, target, sample count, and source.

Re-exports§

pub use boston_housing::BostonHousing;
pub use breast_cancer::BreastCancer;
pub use california_housing::CaliforniaHousing;
pub use diabetes::Diabetes;
pub use iris::Iris;
pub use palmer_penguins::PalmerPenguins;
pub use titanic::Titanic;
pub use wine_quality::red_wine_quality::RedWineQuality;
pub use wine_quality::white_wine_quality::WhiteWineQuality;
pub use wine_recognition::WineRecognition;

Modules§

boston_housing
Boston Housing dataset module.
breast_cancer
Breast Cancer Wisconsin (Diagnostic) dataset module.
california_housing
California Housing dataset module.
diabetes
Diabetes dataset module.
iris
Iris flower dataset module.
palmer_penguins
Palmer Penguins dataset module.
titanic
Titanic dataset module.
wine_quality
Wine Quality dataset module.
wine_recognition
Wine Recognition dataset module.