bigdl.models.local_lenet package

Submodules

bigdl.models.local_lenet.local_lenet module

bigdl.models.local_lenet.local_lenet.build_model(class_num)

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

Get and normalize the mnist data. We would download it automatically if the data doesn’t present at the specific location.

Parameters:

• data_type – training data or testing data
• location – Location storing the mnist

Returns:

(features: Ndarray, label: Ndarray)

Module contents