Skip to content
A PyTorch framework for prediction of tertiary protein structure
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data add sample data Dec 10, 2018
output Prediction example (#12) Jan 29, 2019
pnerf Gpu speed-up (#9) Jan 29, 2019
web-app remove dependency on react-scripts-ts (#7) Jan 26, 2019
.gitignore
LICENSE initial commit Sep 8, 2018
README.md Update README.md Jan 30, 2019
__main__.py
dashboard.py Faster validation (#8) Jan 27, 2019
demo.mpg
examplemodelrun.png
models.py Gpu speed-up (#9) Jan 29, 2019
openprotein.py
prediction.py Update prediction.py Jan 30, 2019
preprocessing.py forgot an import Apr 19, 2019
preprocessing_cli.py added suggestions Apr 19, 2019
start_web_app.sh use npm as default package manager (#6) Jan 26, 2019
util.py

README.md

OpenProtein

A PyTorch framework for tertiary protein structure prediction.

Alt text

Getting started

To run this project, simply git clone the repository and type python __main__.py in the terminal:

$ python __main__.py
------------------------
--- OpenProtein v0.1 ---
------------------------
Live plot deactivated, see output folder for plot.
Starting pre-processing of raw data...
Preprocessed file for testing.txt already exists.
force_pre_processing_overwrite flag set to True, overwriting old file...
Processing raw data file testing.txt
Wrote output to 81 proteins to data/preprocessed/testing.txt.hdf5
Completed pre-processing.
2018-09-27 19:27:34: Train loss: -781787.696391812
2018-09-27 19:27:35: Loss time: 1.8300042152404785 Grad time: 0.5147676467895508
...

Developing a Predictive Model

See models.py for examples of how to create your own model.

Using a Predictive Model

See prediction.py for examples of how to use pre-trained models.

Memory Usage

OpenProtein includes a preprocessing tool (preprocessing.py) which will transform the standard ProteinNet format into a hdf5 file and save it in data/preprocessed/. This is done in a memory-efficient way (line-by-line).

The OpenProtein PyTorch data loader is memory optimized too - when reading the hdf5 file it will only load the samples needed for each minibatch into memory.

License

Please see the LICENSE file in the root directory.

You can’t perform that action at this time.