Struct MnistDataset 

pub struct MnistDataset { /* private fields */ }

Reads MNIST samples from a HuggingFace parquet file.

Implementations

impl MnistDataset

pub fn open(path: impl AsRef<Path>) -> Result<Self>

Open a parquet file and read its metadata.

pub fn len(&self) -> usize

Total number of rows in the dataset.

pub fn is_empty(&self) -> bool

Whether the dataset has zero rows.

pub fn read_batch(&self, offset: usize, count: usize) -> Result<MnistBatch>

Read count consecutive rows starting at offset (0-indexed).

The returned batch is truncated if offset + count > len().