burn-dataset 0.21.0

Library with simple dataset APIs for creating ML data pipelines
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

//! # Dataset Transformations
//!
//! This module provides a collection of [`crate::Dataset`] composition wrappers;
//! providing composition, subset selection, sampling, random shuffling, and windowing.
//!
//! * [`ComposedDataset`] - composes a list of datasets.
//! * [`PartialDataset`] - selects a contiguous index range subset of a dataset.
//! * [`ShuffledDataset`] - a randomly shuffled / mutably shuffle-able dataset;
//!   a thin wrapper around [`SelectionDataset`].
//! * [`SamplerDataset`] - samples a dataset; support for with/without replacement,
//!   and under/oversampling.
//! * [`SelectionDataset`] - selects a subset of a dataset via indices; support for shuffling.
//! * [`WindowsDataset`] - creates a sliding window over a dataset.
mod composed;
mod mapper;
mod options;
mod partial;
mod sampler;
mod selection;
mod shuffle;
mod window;

pub use composed::*;
pub use mapper::*;
pub use options::*;
pub use partial::*;
pub use sampler::*;
pub use selection::*;
pub use shuffle::*;
pub use window::*;