https://medium.com/@monocosmo77/working-with-variational-autoencoders-part5-machine-learning-35df4763d47c