https://omegaxyz.com/2018/05/31/tensorflow-simple-classification/