Imported from https://github.com/samsinai/VAE_protein_function