This sample creates and trains MNIST model in Jupyter Notrbook using Tensorflow and then deploys the trained model to Google Cloud MLE to serve the model, and accept prediction requests by REST API/Google MLE Client API.
refer to requirements.txt
Tensorflow MNIST dataset
- main.py: uses Google Cloud MLE Client API to make prediction calls
- MNIST Notebook.ipynb: A jupyter notebook file used to create and train the model then use gsutil tool to upload the trained model to Google Cloud Storage (Bucket)
- Clone the repository: git clone https://github.com/m0hamdan/Google-CloudML-MNIST-Digits.git
- Run the following command in the repo root directory: python main.py
None
- Tensorflow webiste as I used the model created here https://www.tensorflow.org/tutorials/estimators/cnn