How to use Dataset in TensorFlow – Towards Data Science