Deep learning tools and models for MALDI-TOF spectra analysis
Project description
maldi-nn
Deep learning tools and models for MALDI-TOF mass spectra analysis.
Package features:
- Reading and preprocessing functions for MALDI-TOF MS spectra.
- Model definitions to process SMILES strings with state-of-the-art techniques (for feature-based AMR prediction).
- Model definitions to pre-train state-of-the-art Transformer networks on MALDI-TOF MS data
- Model definitions and scripts to train AMR models on the DRIAMS database.
- Model definitions and scripts to train species identification models.
Install
maldi-nn
is distributed on PyPI.
pip install maldi-nn
You may need to install PyTorch before running this command in order to ensure the right CUDA kernels for your system are installed
Academic Reproducibility
This package contains all code and scripts to reproduce: "An antimicrobial drug recommender system using MALDI-TOF MS and dual-branch neural networks", and "Pre-trained Maldi Transformers improve MALDI-TOF MS-based prediction" (in draft). All information regarding reproducing our results can be found in the reproduce folder README
Credits
- Implementations of many MALDI reading and processing functions were based on the R package MaldiQuant.
- Topological Peak Filtering was taken from the Topf package.
Citation
@article{de2023antimicrobial,
title={An antimicrobial drug recommender system using MALDI-TOF MS and dual-branch neural networks},
author={De Waele, Gaetan and Menschaert, Gerben and Waegeman, Willem},
journal={bioRxiv},
pages={2023--09},
year={2023},
publisher={Cold Spring Harbor Laboratory}
}
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.