Awesome

Main Results

https://pubs.acs.org/doi/10.1021/acs.jcim.9b00387

Training command example

python -u train.py --dropout_rate=0.3 --epoch=1000 --ngpu=1 --batch_size=256 --num_workers=0

We added only about data of 1000 samples in the data folder due to the size of the dataset so the performance is much lower than the paper. Each sample is saved in a pickle file and it consists of two rdkit objects of a ligand and protein. The inputs of the neural network are processed in the dataset class on the fly.