bigdl.models.local_lenet package¶

Submodules¶

bigdl.models.local_lenet.local_lenet.get_mnist(data_type='train', location='/tmp/mnist')[source]¶

Get mnist dataset with features and label as ndarray. Data would be downloaded automatically if it doesn’t present at the specific location.

Parameters:	data_type – “train” for training data and “test” for testing data. location – Location to store mnist dataset.
Returns:	(features: ndarray, label: ndarray)