Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/jaechanglim/GNN_DTI


https://github.com/jaechanglim/GNN_DTI

Last synced: 3 months ago
JSON representation

Awesome Lists containing this project

README

        

![Screenshot](figure.png)

## Main Results
![Screenshot](result.png)

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.