Accessing the features of the auto-encoder model