Compositions from the Novel Materials Discovery (NO-MAD) data set are used in the prediction of our model. This repository contains computational materials science data that is allowed to be curated . For this implementation, we used a specific curated data set of unique reduced chemical formulae . This data was restricted to density functional theory (DFT) calculations and does not include noble gases or radioactive elements. It is also directly usable with the pymatgen.core.Composition class , which is what this implementation exploits.