#! /usr/bin/python
# -*- coding: utf-8 -*-
from tensorlayerx.files.utils import _load_mnist_dataset
from tensorlayerx import logging
logging.set_verbosity(logging.INFO)
__all__ = ['load_mnist_dataset']
[docs]def load_mnist_dataset(shape=(-1, 784), path='data'):
"""Load the original mnist.
Automatically download MNIST dataset and return the training, validation and test set with 50000, 10000 and 10000 digit images respectively.
Parameters
----------
shape : tuple
The shape of digit images (the default is (-1, 784), alternatively (-1, 28, 28, 1)).
path : str
The path that the data is downloaded to.
Returns
-------
X_train, y_train, X_val, y_val, X_test, y_test: tuple
Return splitted training/validation/test set respectively.
Examples
--------
>>> X_train, y_train, X_val, y_val, X_test, y_test = tlx.files.load_mnist_dataset(shape=(-1,784), path='datasets')
>>> X_train, y_train, X_val, y_val, X_test, y_test = tlx.files.load_mnist_dataset(shape=(-1, 28, 28, 1))
"""
logging.info("If can't download this dataset automatically, "
"please download it from the official website manually."
"mnist Dataset <http://yann.lecun.com/exdb/mnist/>."
"Please place dataset under 'data/mnist/' by default.")
return _load_mnist_dataset(shape, path, name='mnist', url='http://yann.lecun.com/exdb/mnist/')