Tensorflow implementation of the VAE with VampPrior paper (https://arxiv.org/abs/1705.07120) for the Machine Learning, Advanced Course at KTH
The model can be trained and used on 3 different datasets:
- Clone the repo
- Install python packages (listed in the
requirements.txt
file)(or alternatevily with conda)pip install -r requirements.txt
Information about the usage can be obtained by running python main.py -h
.
Here are the results obtained with stardard vae on the Frey dataset
Implemented by
- Francesco Zappia (Standard VAE)
- Vittorio Zampinetti (VampPrior)
- Marco Schouten (HVAE)
More generally each of the authors influenced and contributed on each other's work.
Part of the implementation is inspired by
Distributed under the MIT License. See LICENSE
for more information.