/Mol-GenDA

Primary LanguagePython

StructureGeneration

JT-VAE

GAN

  • Generator

one-shot GenDA

  • Domain adaption

Usage

  • In ./GAN

    • "sh train.sh" train the GAN with the whole dataset ZINC-250K

    • "sh finetune.sh" finetune the GAN with specific dataset so that it can generate materials similar to the target

    • "sh generate.sh" generate the desired material

  • In ./GEN

    • "sh decode.sh" docode the material SMILEs from latent vector
  • In ./Valid

    • "sh valid.sh" validate the results