mnist3D

fun mnist3D(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>

Loads the MNIST 3D dataset. This is a dataset of 10,000 16x16x16 grayscale 3D images of the 10 digits, along with a test set of 2,000 3D images.

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. MNIST 3D dataset was created by daavoo as a transformation of original MNIST dataset to 3D images to provide simple example of working with 3D images.

Return

Train and test datasets. Each dataset includes X and Y data. X data are float arrays of grayscale image data with shapes (num_samples, 16, 16, 16). Y data float arrays of digit labels (integers in range 0-9) with shapes (num_samples,).

Parameters

cacheDirectory

Cache directory to cached models and datasets.