DiffSBDD icon indicating copy to clipboard operation
DiffSBDD copied to clipboard

How to sample mols on all pockets of the test set of Crossdocked dataset?

Open youmingzhao91 opened this issue 2 years ago • 7 comments

Hi, I am trying to set DiffSBDD cond method as one of baselines in my work, I would like to sample mols given 100 pockets from the test set of Crossdocked dataset on my own machine. What can I do?

python generate_ligands.py <checkpoint>.ckpt --pdbfile <pdb_file>.pdb --outdir <output_dir> --resi_list <list_of_pocket_residue_ids>

How can I specify --resi_list <list_of_pocket_residue_ids> or --ref_ligand :?

Does the following command only work for the Bindingmoad dataset? I only need to run your sample code on the test set of Crossdoced.

python test.py <checkpoint>.ckpt --test_dir <bindingmoad_dir>/processed_noH/test/ --outdir <output_dir> --fix_n_nodes

Could you please give me some instructions for my goal?

youmingzhao91 avatar May 30 '23 16:05 youmingzhao91

In addition, I have preprocessed the CrossDocked dataset in the same way as Pocket2Mol.

youmingzhao91 avatar May 30 '23 17:05 youmingzhao91

Thanks for your interest in our work.

The following command would work for crossdocked dataset as well for sampling over the test set

python test.py <checkpoint>.ckpt --test_dir <bindingmoad_dir>/processed_noH/test/ --outdir <output_dir> --fix_n_nodes

yuanqidu avatar May 31 '23 18:05 yuanqidu

Thanks! I'm using generate_ligands.py with a for-loop to do that. It works now.

youmingzhao91 avatar Jun 01 '23 02:06 youmingzhao91

Hi, I would just like to add that this repository only contains checkpoints for BindingMOAD at the moment. Please keep this in mind when sampling molecules for the CrossDocked test set.

arneschneuing avatar Jun 01 '23 08:06 arneschneuing

Where can I get the checkpoint for the CrossDocked dataset? Can you upload it here?

youmingzhao91 avatar Jun 01 '23 08:06 youmingzhao91

Hi, I would just like to add that this repository only contains checkpoints for BindingMOAD at the moment. Please keep this in mind when sampling molecules for the CrossDocked test set.

The following checkpoint

https://github.com/arneschneuing/DiffSBDD/blob/main/checkpoints/full_atom.ckpt

is only for BindingMOAD? However, I am using it to sample molecules for the CrossDocked test set right now.

youmingzhao91 avatar Jun 01 '23 08:06 youmingzhao91

Yes, the model was trained on the Binding MOAD dataset

arneschneuing avatar Jun 09 '23 07:06 arneschneuing