ceras::dataset::mnist Namespace Reference

Functions
auto	load_data (std::string const &path=std::string{"./dataset/mnist"})

Function Documentation

load_data()

auto ceras::dataset::mnist::load_data ( std::string const &  path = std::string{"./dataset/mnist"} )

inline

Loads the MNIST dataset.

Parameters

path	Path where to cache the dataset locally. Default to "./dataset/mnist", should be updated if running the program somewhere else.

Returns

A tuple of 4 tensors: x_train, y_train, x_test, y_test. x_train, x_test: uint8 arrays of grayscale image data with shapes (num_samples, 28, 28). y_train, y_test: uint8 tensor of digit labels (integers in range 0-9) with shapes (num_samples, 10). Note: for digit 0, the corresponding array is [[1, 0, 0, 0, 0, 0, 0, 0, 0, 0]].

Example usage:

auto const& [x_train, y_train, x_test, y_test] = ceras::dataset::mnist::load_data("/home/feng/dataset/mnist");

Yann LeCun and Corinna Cortes hold the copyright of MNIST dataset, which is available under the terms of the Creative Commons Attribution-Share Alike 3.0 license.