Autoencoder For Image Classification Python