Struct MiniBatchSampler 

pub struct MiniBatchSampler { /* private fields */ }

Mini-batch sampler that yields batches from a dataset.

Construct via MiniBatchSampler::new and iterate with iter_batches.

Implementations

impl MiniBatchSampler

pub fn new(config: SamplerConfig) -> Self

Create a new sampler with the given configuration.

pub fn config(&self) -> &SamplerConfig

Return a reference to the current configuration.

pub fn iter_batches( &self, data: &[Vec<f64>], labels: &[usize], ) -> Result<Vec<MiniBatch>>

Generate all mini-batches for the given data and labels.

Returns a Vec<MiniBatch> where each batch has at most config.batch_size samples. When drop_last is true, the final partial batch is omitted.

Errors

Returns an error if data and labels have different lengths, or if batch_size is zero.

Trait Implementations

impl Clone for MiniBatchSampler

fn clone(&self) -> MiniBatchSampler

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MiniBatchSampler

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.