theislab / cpa

The Compositional Perturbation Autoencoder (CPA) is a deep generative framework to learn effects of perturbations at the single-cell level. CPA performs OOD predictions of unseen combinations of drugs, learns interpretable embeddings, estimates dose-response curves, and provides uncertainty estimates.
BSD 3-Clause "New" or "Revised" License
83 stars 17 forks source link

Added custom latent/expression reconstruction functionality. #28

Closed ArianAmani closed 1 year ago

ArianAmani commented 1 year ago

Added function custom_predict that allows you to predict the output of the model (latents and the gene expression reconstruction) with respect to only the specific covariate keys, given to the function.