bigdl.models.lenet package

Submodules

bigdl.models.lenet.lenet5 module

bigdl.models.lenet.lenet5.build_model(class_num)[source]
bigdl.models.lenet.lenet5.get_mnist(sc, data_type='train', location='/tmp/mnist')[source]

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

Parameters:
  • sc – SparkContext
  • data_type – training data or testing data
  • location – Location storing the mnist
Returns:

A RDD of (features: Ndarray, label: Ndarray)

Module contents