torchvision: mnist_dataset – R documentation

Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!

mnist_dataset

MNIST dataset

Prepares the MNIST dataset and optionally downloads it.

mnist_dataset(
  root,
  train = TRUE,
  transform = NULL,
  target_transform = NULL,
  download = FALSE
)

`root`	(string): Root directory of dataset where `MNIST/processed/training.pt` and `MNIST/processed/test.pt` exist.
`train`	(bool, optional): If True, creates dataset from `training.pt`, otherwise from `test.pt`.
`transform`	(callable, optional): A function/transform that takes in an PIL image and returns a transformed version. E.g, `transforms.RandomCrop`.
`target_transform`	(callable, optional): A function/transform that takes in the target and transforms it.
`download`	(bool, optional): If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again.

Models, Datasets and Transformations for Images

v0.3.0

MIT + file LICENSE

Authors

Daniel Falbel [aut, cre], RStudio [cph]

Initial release