Documentation | External Resources | Datasets | Examples
ChemicalX is a deep learning library for drug-drug interaction, polypharmacy side effect and synergy prediction. The library consists of data loaders and integrated benchmark datasets. It also includes state-of-the-art deep neural network architectures that solve the drug pair scoring task. Implemented methods cover traditional SMILES string based techniques and neural message passing based models.
Drug Pair Scoring Explained
Our framework solves the so called drug pair scoring task of computational chemistry. In this task a machine learning model has to predict the outcome of administering two drugs together in a biological or chemical context. Deep learning models which solve this task have an architecture with two distinctive parts:
- A drug encoder layer which takes a pair of drugs as an input (blue and red drugs below).
- A head layer which outputs scores in the administration context - polypharmacy in our explanatory figure.
Case Study Tutorials
We provide in-depth case study tutorials in the Documentation, each covers an aspect of ChemicalX’s functionality.
Citing
If you find ChemicalX and the new datasets useful in your research, please consider adding the following citation:
@inproceedings{chemicalx,
author = {Benedek Rozemberczki and Charles Tapley Hoyt and Benjamin Gyori},
title = {{ChemicalX: A Deep Learning Library fo Drug Pair Scoring}},
year = {2022},
}
A simple example
Methods Included
In detail, the following temporal graph neural networks were implemented.
2017
- DeepCCI from DeepCCI: End-to-end Deep Learning for Chemical-Chemical Interaction Prediction (ACM BCB)
2018
-
DeepDDI from Deep Learning Improves Prediction of Drug–Drug and Drug–Food Interactions (PNAS)
-
DeepSynergy from DeepSynergy: Predicting Anti-Cancer Drug Synergy with Deep Learning (Bioinformatics)
2019
-
MR-GNN from MR-GNN: Multi-Resolution and Dual Graph Neural Network for Predicting Structured Entity Interactions (IJCAI)
-
MHCADDI from Drug-Drug Adverse Effect Prediction with Graph Co-Attention (ICML)
2020
-
CASTER from CASTER: Predicting Drug Interactions with Chemical Substructure Representation (AAAI)
-
SSI-DDI from SSI–DDI: Substructure–Substructure Interactions for Drug–Drug Interaction Prediction (Briefings in Bioinformatics)
-
EPGCN-DS from Structure-Based Drug-Drug Interaction Detection via Expressive Graph Convolutional Networks and Deep Sets (AAAI)
-
AuDNNSynergy from Synergistic Drug Combination Prediction by Integrating Multiomics Data in Deep Learning Models (Methods in Molecular Biology)
-
DeepDrug from DeepDrug: A General Graph-Based Deep Learning Framework for Drug Relation Prediction (PMC)
-
GCN-BMP from GCN-BMP: Investigating graph representation learning for DDI prediction task (Methods)
2021
-
DPDDI from DPDDI: a Deep Predictor for Drug-Drug Interactions (BMC Bioinformatics)
-
DeepDDS from DeepDDS: Deep Graph Neural Network with Attention Mechanism to Predict Synergistic Drug Combinations (Briefings in Bioinformatics)
-
MatchMaker from MatchMaker: A Deep Learning Framework for Drug Synergy Prediction (ACM TCBB)
Auxiliary Layers
Head over to our documentation to find out more about installation, creation of datasets and a full list of implemented methods and available datasets. For a quick start, check out the examples in the examples/ directory.
If you notice anything unexpected, please open an issue. If you are missing a specific method, feel free to open a feature request.
Installation
Binaries are provided for Python version <= 3.9.
PyTorch 1.9.0
To install the binaries for PyTorch 1.9.0, simply run
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html
pip install torch-geometric
pip install chemicalx
where ${CUDA} should be replaced by either cpu, cu102, or cu111 depending on your PyTorch installation.
cpu |
cu102 |
cu111 |
|
|---|---|---|---|
| Linux |
|
|
|
| Windows |
|
|
|
| macOS |
|
Expand to see installation guides for older PyTorch versions...
PyTorch 1.8.0
To install the binaries for PyTorch 1.8.0, simply run
$ pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+${CUDA}.html
$ pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.0+${CUDA}.html
$ pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.8.0+${CUDA}.html
$ pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.8.0+${CUDA}.html
$ pip install torch-geometric
$ pip install torch-geometric-temporal
where ${CUDA} should be replaced by either cpu, cu101, cu102, or cu111 depending on your PyTorch installation.
cpu |
cu101 |
cu102 |
cu111 |
|
|---|---|---|---|---|
| Linux |
|
|
|
|
| Windows |
|
|
|
|
| macOS |
|
PyTorch 1.7.0
To install the binaries for PyTorch 1.7.0, simply run
$ pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.7.0.html
$ pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.7.0.html
$ pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.7.0.html
$ pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.7.0.html
$ pip install torch-geometric
$ pip install torch-geometric-temporal
where ${CUDA} should be replaced by either cpu, cu92, cu101, cu102 or cu110 depending on your PyTorch installation.
cpu |
cu92 |
cu101 |
cu102 |
cu110 |
|
|---|---|---|---|---|---|
| Linux |
|
|
|
|
|
| Windows |
|
|
|
|
|
| macOS |
|
PyTorch 1.6.0
To install the binaries for PyTorch 1.6.0, simply run
$ pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-geometric
$ pip install torch-geometric-temporal
where ${CUDA} should be replaced by either cpu, cu92, cu101 or cu102 depending on your PyTorch installation.
cpu |
cu92 |
cu101 |
cu102 |
|
|---|---|---|---|---|
| Linux |
|
|
|
|
| Windows |
|
|
|
|
| macOS |
|
Running tests
$ python setup.py test
License
59 Dec 09, 2022
1 Nov 09, 2021
2 Jan 08, 2022
122 Dec 28, 2022
113 Dec 27, 2022
589 Dec 30, 2022
621 Jan 08, 2023
56 Dec 16, 2022
2 Dec 08, 2022
35 Jan 04, 2023
6 Jan 17, 2022
2.9k Jan 07, 2023
532 Dec 29, 2022
625 Oct 27, 2022
45 Nov 21, 2022
98 Dec 21, 2022
77 Nov 11, 2022
1.7k Jan 01, 2023
98 Dec 28, 2022
8.6k Jan 04, 2023