search

pascalnotin / uncertainty_guided_optimization

Official repository for the paper "Improving black-box optimization in VAE latent space using decoder uncertainty" (Pascal Notin, José Miguel Hernández-Lobato, Yarin Gal)
MIT License
29 stars 5 forks source link

Reproducing 3-digit MNIST BO task with decoder uncertainty #2

Open PraljakReps opened 1 year ago

PraljakReps commented 1 year ago

Hello @pascalnotin,

Excellent paper and repository; however, I noticed that this repository only contains code on the junction tree VAE (JTVAE).

Is there reproducible code for the more straightforward task (MNIST 3-digit generation) which was presented in the paper: Improving black-box optimization in VAE latent space using decoder uncertainty (Pascal Notin, José Miguel Hernández-Lobato, Yarin Gal)?

pascalnotin commented 1 year ago

Hi @PraljakReps, Thanks for the kind words! Will try and upload another task to the repo when I get the chance (post NeurIPS). Best, Pascal