Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -44,12 +44,23 @@ sudo make install
cd ..
```

Or easily setup by using install.sh:
```
chmod +x install.sh
./install.sh
```

Usage example
---------------

```
python predict.py -p protein.pdb -mp model_path -o output_path
```

Ex:
```
python predict.py -p protein.pdb -mp models/ -o output
```

For more input options, check 'predict.py'. All other molecules (waters, ions, ligands) should be removed from the structure. If the input protein has not been protonated, add --protonate to the execution command.\
The provided models have been trained on a subset of scPDB (training_subset_of_scpdb.proteins)
31 changes: 31 additions & 0 deletions install.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
conda create -n deepsurf python=3.6.9
conda activate deepsurf
sudo apt update
sudo apt install -y p7zip
sudo apt install -y libopenbabel-dev
sudo apt install -y g++
sudo apt install -y swig
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
conda install -c conda-forge openbabel
pip install tensorflow-gpu==1.13.1
pip install scikit-learn==0.20.3

git clone https://github.com/stemylonas/DeepSurf
cd DeepSurf

cd lds
chmod a+x compile.sh
./compile.sh
cd ..

pip install gdown --ignore-installed certifi
gdown 1nIBoD3_5nuMqgRGx4G1OHZwLsiUjb7JG
p7zip -d models.7z

wget www.cgl.ucsf.edu/Overview/ftp/dms.zip
unzip dms.zip
rm dms.zip
cd dms
chmod 777 -R .
sudo make install
cd ..
2 changes: 1 addition & 1 deletion protein.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
"""

import os, numpy as np
import pybel
from openbabel import pybel
from utils import simplify_dms


Expand Down
6 changes: 3 additions & 3 deletions tfbio_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

import numpy as np

import pybel
from openbabel import pybel
from math import ceil, sin, cos, sqrt, pi
from itertools import combinations
import collections
Expand Down Expand Up @@ -62,7 +62,7 @@ def __init__(self, atom_codes=None, atom_labels=None,
specified this argument is ignored.
named_properties: list of strings, optional
Names of atomic properties to retrieve from pybel.Atom object. If
not specified ['hyb', 'heavyvalence', 'heterovalence',
not specified ['hyb', 'heavydegree', 'heterodegree',
'partialcharge'] is used.
save_molecule_codes: bool, optional (default True)
If set to True, there will be an additional feature to save
Expand Down Expand Up @@ -151,7 +151,7 @@ def __init__(self, atom_codes=None, atom_labels=None,
self.NAMED_PROPS = named_properties
else:
# pybel.Atom properties to save
self.NAMED_PROPS = ['hyb', 'heavyvalence', 'heterovalence',
self.NAMED_PROPS = ['hyb', 'heavydegree', 'heterodegree',
'partialcharge']
self.FEATURE_NAMES += self.NAMED_PROPS

Expand Down