Autoencoder Neural Network Python Code