Python Data Augmentation Tensorflow Playground