This code generates all results in our manuscript:
@misc{stirn2024vampprior,
title={The VampPrior Mixture Model},
author={Andrew Stirn and David A. Knowles},
year={2024},
eprint={2402.04412},
archivePrefix={arXiv},
primaryClass={cs.LG}
}
Our Dockerfile
creates the computational environment from TensorFlow's official image for version 2.12.0 with GPU support and uses requirements.txt
to install additionally required packages.
Run elbo_surgery_run.sh
to produce Table 1.
Run clustering_experiments_run.sh
to produce Table 2 and Figures 1, 2, and 4.
Run single_cell_experiments_run.sh
to produce Tables 3 and 4 and Figures 3, 5-8.
priors.py
implements the DP GMM and VMM.
models.py
has the VAE models.
single_cell_models.py
has our implementation of scVI (Lopez et al., 2018).
clustering_*.py
are used by clustering_experiments_run.sh
.
elbo_surgery_*.py
are used by elbo_surgery_run.sh
.
single_cell_*.py
are used by single_cell_experiments_run.sh
.