Each frame of the aspirin trajectory contained the molecule with its atoms labeled by atomic number and specified in Cartesian coordinates. I transferred this information into a new data structure, augmenting it with embedded chemical inputs. For the graph-based models, I determined the connectivity of each atom using a cutoff function and constructed a graph for each molecule, representing atoms as nodes and bonds as edges.