Struct burn::data::dataset::vision::MnistDataset

pub struct MnistDataset { /* private fields */ }

The MNIST dataset consists of 70,000 28x28 black-and-white images in 10 classes (one for each digits), with 7,000 images per class. There are 60,000 training images and 10,000 test images.

The data is downloaded from the web from the CVDF mirror.

Implementations

impl MnistDataset

pub fn train() -> MnistDataset

Creates a new train dataset.

pub fn test() -> MnistDataset

Creates a new test dataset.

Trait Implementations

impl Dataset<MnistItem> for MnistDataset

fn get(&self, index: usize) -> Option<MnistItem>

Gets the item at the given index.

fn len(&self) -> usize

Gets the number of items in the dataset.

fn is_empty(&self) -> bool

Checks if the dataset is empty.

fn iter(&self) -> DatasetIterator<'_, I>

where Self: Sized,

Returns an iterator over the dataset.

fn windows(&self, size: usize) -> WindowDataset<'_, I>

where Self: Sized,

Returns a new Dataset of all the windows of length size. The windows overlap. Is empty if the input Dataset is shorter than size.