mnist
fun mnist(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>
Content copied to clipboard
Loads the MNIST dataset. This is a dataset of 60,000 28x28 grayscale images of the 10 digits, along with a test set of 10,000 images. More info can be found at the MNIST homepage.
NOTE: Yann LeCun and Corinna Cortes hold the copyright of MNIST dataset, which is a derivative work from original NIST datasets. MNIST dataset is made available under the terms of the Creative Commons Attribution-Share Alike 3.0 license.
Return
Train and test datasets. Each dataset includes X and Y data. X data are uint8 arrays of grayscale image data with shapes (num_samples, 28, 28). Y data uint8 arrays of digit labels (integers in range 0-9) with shapes (num_samples,).
Parameters
cacheDirectory
Cache directory to cached models and datasets.