In this repository, we show how to download and apply the Misato database for AI models. You can access the calculated properties of different protein-ligand structures and use them for training in Pytorch based dataloaders. We provide a small sample of the dataset along with the repo.
You can freely download the FULL MISATO dataset from Zenodo using the links below:
Understand the structure of our dataset and how to access each molecule's properties.
Load the PyTorch Dataloaders of each dataset.
Load the PyTorch lightning Datamodules of each dataset.
πΒ Β Quickstart
We recommend to pull our MISATO image from DockerHub or to create your own image (see docker/). The images use cuda version 11.8. We recommend to install on your own system a version of CUDA that is a least 11.8 to ensure that the drivers work correctly.
# clone project
git clone https://github.com/t7morgen/misato-dataset.git
cd misato-dataset
For singularity use:
# get the container image
singularity pull docker://sab148/misato-dataset
singularity shell misato.sif
βββ data <- Project data
β βββMD
β β βββ h5_files <- storage of dataset
β β βββ splits <- train, val, test splits
β βββQM
β β βββ h5_files <- storage of dataset
β β βββ splits <- train, val, test splits
β
βββ src <- Source code
β βββ data
β β βββ components <- Datasets and transforms
β β βββ md_datamodule.py <- MD Lightning data module
β β βββ qm_datamodule.py <- QM Lightning data module
β β β
β β βββ processing <- Skripts for preprocessing, inference and conversion
β β βββ...
β βββ getting_started.ipynb <- notebook : how to load data and interact with it
β βββ inference.ipynb <- notebook how to run inference
β
βββ docker <- Dockerfile and execution script
βββ README.md
Installation using your own conda environment
In case you want to use conda for your own installation please create a new misato environment.
In order to install pytorch geometric we recommend to use pip (within conda) for installation and to follow the official installation instructions:pytorch-geometric/install
Depending on your CUDA version the instructions vary. We show an example for the CUDA 11.8.
If you found this work useful please consider citing the article.
@article{siebenmorgen2024misato,
title={MISATO: machine learning dataset of protein--ligand complexes for structure-based drug discovery},
author={Siebenmorgen, Till and Menezes, Filipe and Benassou, Sabrina and Merdivan, Erinc and Didi, Kieran and Mour{\~a}o, Andr{\'e} Santos Dias and Kitel, Rados{\l}aw and Li{\`o}, Pietro and Kesselheim, Stefan and Piraud, Marie and Theis, Fabian J. and Sattler, Michael and Popowicz, Grzegorz M.},
journal={Nature Computational Science},
pages={1--12},
year={2024},
publisher={Nature Publishing Group US New York}
}
Ecosystem Role
Standard MoltPulse indexed agent.
Embed Badge
Show off your Pulse Score in your GitHub README to build trust and rank higher.