Struct MapDataset 

pub struct MapDataset<D: Dataset> { /* private fields */ }

Wraps a dataset and applies a Transform lazily on each get().

Implementations

impl<D: Dataset> MapDataset<D>

pub fn new( inner: D, transform: Box<dyn Transform>, feat_shape: Vec<usize>, tgt_shape: Vec<usize>, ) -> Self

Create a MapDataset.

feat_shape and tgt_shape describe the shapes after the transform is applied. If the transform doesn’t change shapes, you can clone them from the inner dataset.

pub fn same_shape(inner: D, transform: Box<dyn Transform>) -> Self

Convenience: create a MapDataset whose shapes are unchanged.

Trait Implementations

impl<D: Dataset> Dataset for MapDataset<D>

fn len(&self) -> usize

Total number of samples in the dataset.

fn get(&self, index: usize) -> Sample

Retrieve the sample at position index.

fn feature_shape(&self) -> &[usize]

The shape of a single feature sample (without batch dim).

fn target_shape(&self) -> &[usize]

The shape of a single target sample (without batch dim).

fn name(&self) -> &str

Optional human-readable name.

fn is_empty(&self) -> bool

Whether the dataset is empty.