Steps of reproduction:

  • Download the datasets from the link given in MoleculeNet (QM9) and ConfVAE(GEOM-QM9 & GEOM-Drugs)
  • Run script_conf_feature.py to get the hyper-parameters $\lambda$
  • Copy the xxx.slurm from slurms/ and run it