An extension of Spike-Slab Variational Autoencoder for causal representation learning with learnable interventions.
Download or clone this repository. Then from inside the folder simply run:
pip install -e .
The jupyter notebooks of this implementation were tested with the packages of the following version numbers:
- python == 3.8
- scvi-tools == 0.18.0
- torchmetrics == 0.11.4
- pandas == 1.5.3
This work is based on the implementation of sVAE+ by Lopez et al. The original implementation can be found at: https://github.com/Genentech/sVAE/tree/main