Struct DataLoaderBuilder 

pub struct DataLoaderBuilder { /* private fields */ }

Builder for creating DataLoaders with more complex configurations.

Implementations

impl DataLoaderBuilder

pub fn new() -> Self

Creates a new builder with default values.

pub fn batch_size(self, size: usize) -> Self

Sets the batch size.

pub fn shuffle(self, shuffle: bool) -> Self

Sets whether to shuffle.

pub fn drop_last(self, drop_last: bool) -> Self

Sets whether to drop the last incomplete batch.

pub fn seed(self, seed: u64) -> Self

Sets the random seed.

pub fn build<D: Dataset>(self, dataset: D) -> Result<DataLoader<D>>

Builds a DataLoader with the given dataset.

Errors

Returns an error if the batch size is zero.

Trait Implementations

impl Debug for DataLoaderBuilder

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

Formats the value using the given formatter.

impl Default for DataLoaderBuilder

fn default() -> DataLoaderBuilder

Returns the “default value” for a type.