Skip to content
OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research
Python Other
  1. Python 98.5%
  2. Other 1.5%
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.
benchmark_datasets added tox21 and logP datasets Aug 9, 2018
devtools Added devtools Feb 19, 2019
docs
example_configs freated folder with example configs Aug 9, 2018
openchem Added MLP instantiation test to openchem modules unit tests Feb 19, 2019
.gitignore Modified .gitignore file Feb 19, 2019
.travis.yml Fixed path to the folder with unit tests Feb 19, 2019
Dockerfile Added Dockerfile and requirements Aug 10, 2018
LICENSE Create LICENSE Aug 10, 2018
README.md updated documentation address Aug 13, 2018
launch.py Update launch.py Aug 9, 2018
logp_gcnn_config.py
requirements.txt Added pytest and pytest-cov to the requirements Feb 19, 2019
run.py added random seed Aug 9, 2018
setup.cfg Added setup files Feb 19, 2019
setup.py Added setup files Feb 19, 2019
tox21_rnn_config.py Update tox21_rnn_config.py Aug 15, 2018
versioneer.py Added setup files Feb 19, 2019

README.md

OpenChem

OpenChem

OpenChem is a deep learning toolkit for Computational Chemistry with PyTorch backend. The goal of OpenChem is to make Deep Learning models an easy-to-use tool for Computational Chemistry and Drug Design Researchers.

Main features

  • Modular design with unified API, modules can be easily combined with each other.
  • OpenChem is easy-to-use: new models are built with only configuration file.
  • Fast training with multi-gpu support.
  • Utilities for data preprocessing.
  • Tensorboard support.

Documentation

Check out OpenChem documentation here.

Supported functionality

Tasks:

  • Classification (binary or multi-class)
  • Regression
  • Multi-task (such as N binary classification tasks)

Data types

  • Sequences of characters such as SMILES strings or amino-acid sequences
  • Molecular graphs. OpenChem takes care of converting SMILES strings into molecular graphs

Modules:

  • Token embeddings
  • Recurrent neural network encoders
  • Graph convolution neural network encoders
  • Multi-layer perceptrons

We are working on populating OpenChem with more models and other building blocks.

Installation

Requirements

In order to get started you need:

General installation

If you installed your Python with Anacoda you can run the following commands to get started:

git clone https://github.com/Mariewelt/OpenChem.git
cd Openchem
conda install --yes --file requirements.txt
conda install -c rdkit rdkit nox cairo
conda install pytorch torchvision -c pytorch
pip install tensorflow-gpu

If your CUDA version is other than 9.0, check Pytorch and Tensorflow websites for different installation instructions.

Installation with Docker

Alternative way of installation is with Docker. We provide a Dockerfile, so you can run your models in a container that already has all the necessary packages installed. You will also need nvidia-docker in order to run models on GPU.

Acknowledgements

OpenChem is sponsored by the University of North Carolina at Chapel Hill and NVIDIA Corp.

UNC NVIDIA

You can’t perform that action at this time.