Simple machine learning in Python/Tensorflow with model saving - View it on GitHub