Struct burn::data::dataset::vision::MnistDataset
pub struct MnistDataset { /* private fields */ }Expand description
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
impl MnistDataset
pub fn train() -> MnistDataset
pub fn train() -> MnistDataset
Creates a new train dataset.
pub fn test() -> MnistDataset
pub fn test() -> MnistDataset
Creates a new test dataset.
Trait Implementations§
§impl Dataset<MnistItem> for MnistDataset
impl Dataset<MnistItem> for MnistDataset
§fn iter(&self) -> DatasetIterator<'_, I> ⓘwhere
Self: Sized,
fn iter(&self) -> DatasetIterator<'_, I> ⓘwhere
Self: Sized,
Returns an iterator over the dataset.
Auto Trait Implementations§
impl Freeze for MnistDataset
impl RefUnwindSafe for MnistDataset
impl Send for MnistDataset
impl Sync for MnistDataset
impl Unpin for MnistDataset
impl UnwindSafe for MnistDataset
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
source§impl<T> IntoEither for T
impl<T> IntoEither for T
source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moresource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more§impl<T> Pointable for T
impl<T> Pointable for T
§impl<I, T> Windows<I> for Twhere
T: Dataset<I>,
impl<I, T> Windows<I> for Twhere
T: Dataset<I>,
§fn windows(&self, size: usize) -> WindowsIterator<'_, I> ⓘ
fn windows(&self, size: usize) -> WindowsIterator<'_, I> ⓘ
Is empty if the Dataset is shorter than size.
§Panics
Panics if size is 0.
§Examples
use crate::burn_dataset::{
transform::{Windows, WindowsDataset},
Dataset, InMemDataset,
};
let items = [1, 2, 3, 4].to_vec();
let dataset = InMemDataset::new(items.clone());
for window in dataset.windows(2) {
// do sth with window
}