Notebook for my medium article about how to use Dataset API in TensorFlow - View it on GitHub