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
|
---|---|
train | (bool, optional): If True, creates dataset from
|
transform | (callable, optional): A function/transform that takes in an
PIL image and returns a transformed version. E.g,
|
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. |