https://github.com/tensorflow/models/blob/master/samples/outreach/blogs/blog_estimators_dataset.py