torchdrug
torchdrug copied to clipboard
Pipeline to get graph encoding from a list of smiles
Hi, I would like to know (if it is possible) how to get the graph from a list of smiles and then feed it into a GNN ( GIN for example) to get their embedding ? Thanks
Yes. You may use data.PackedMolecule.from_smiles
to create a batch of graph from a list of smiles. Check the documentation here.