https://medium.com/@monocosmo77/working-with-variational-autoencoders-part6-machine-learning-5257634b5669