Skip to content

PerturbNet is a deep generative model that can predict the distribution of cell states induced by chemical or genetic perturbation

License

Notifications You must be signed in to change notification settings

welch-lab/PerturbNet

Repository files navigation

PerturbNet

PerturbNet is a deep generative model that can predict the distribution of cell states induced by chemical or genetic perturbation.You can refer our paper PerturbNet predicts single-cell responses to unseen chemical and genetic perturbations for more details

System Requirements and Installation

The current version of PerturbNet requires Python 3.7. All required dependencies are listed in requirements.txt. We recommend creating a clean Conda environment using the following command:

conda create -n "PerturbNet" python=3.7

After setting up the environment, you can install the package by running:

conda activate PerturbNet
pip install --upgrade PerturbNet

We used cuDNN 8.7.0 (cudnn/11.7-v8.7.0) and CUDA 11.7.1 for model training.

We also provide an updated version that removes the dependency on TensorFlow by using Python 3.10. To install:

conda create -n "PerturbNet" python=3.10
conda activate PerturbNet
pip install pip install PerturbNet==0.0.3b1

For reproducibility, we currently recommend using the stable version with Python 3.7.

Core Repository Structure

./perturbnet contains the core modules to train and benchmark the PerturbNet framework.

./perturbnet/net2net contains the conditional invertible neural network (cINN) modules in the GitHub repository of Network-to-Network Translation with Conditional Invertible Neural Networks.

./perturbnet/pytorch_scvi contains our adapted modules to decode latent representations to expression profiles based on scVI version 0.7.1.

Tutorial and Reproducibility

The [./notebooks] directory contains Jupyter notebooks demonstrating how to use PerturbNet and includes code to reproduce the results:

The required data, toy examples, and model weights can be downloaded from Hugging Face.

Reference

If you find our work useful, please consider citing our paper: https://doi.org/10.1038/s44320-025-00131-3.

We appreciate your interest in our work.

About

PerturbNet is a deep generative model that can predict the distribution of cell states induced by chemical or genetic perturbation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •