Using Autoencoder For Feature Extraction Python