Machine Learning for Mobile

Creating and Saving the TF model

First, we first create a simple model and save its computation graph as a serialized GraphDef file. After training the model, we then save the values of its variables into a checkpoint file. We have to turn these two files into an optimized standalone file, which is all we need to use inside the Android app.

For this tutorial, we create a very simple TensorFlow graph that implements a small use case that will calculate (a+b)²=c. Here, we are saving the input as a and b, and the output as c.

To implement this sample program, we are going to use Python. So, as a prerequisite, you need to install python in your machine and install the TensorFlow libraries on your machine using pip.

Please check the software installations/appendix ...