Formation energy value was also concatenated along with the electron density maps. They used VAE along with GAN to generate crystal structures of different classes of materials. However, they needed to use a combination of UNet semantic segmentation network and morphological transformations to convert this representation back to a crystal structure.