├── .gitignore ├── LICENSE ├── MNIST ├── data │ └── process_amat.py ├── params │ ├── gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt │ ├── gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt.meta │ ├── gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt │ ├── gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt.meta │ ├── gauss_regVae_params_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.ckpt │ ├── gauss_regVae_params_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.ckpt.meta │ └── mixVAE_params.pkl ├── samples │ ├── gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png │ ├── gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png │ ├── gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png │ ├── gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png │ └── gauss_regVae_samples_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.png └── train_logs │ ├── dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_2.txt │ ├── dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_3.txt │ ├── dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_4.txt │ ├── dlgmm_trainResults_K_3_hidden_d_200_latent_d_20_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05.txt │ ├── gaussMM_vae_trainResults_K_3_hidden_d_1000_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_1.txt │ ├── gaussMM_vae_trainResults_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt │ ├── gaussMM_vae_trainResults_K_5_hidden_d_1500_latent_d_25_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_1.txt │ ├── gaussMM_vae_trainResults_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt │ ├── gaussMM_vae_trainResults_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt │ └── gauss_regVae_trainResults_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.txt ├── README.md ├── gaussMMVAE.jl ├── knn_mnist.py ├── models ├── __init__.py ├── gaussMMVAE.jl ├── gaussMMVAE_collapsed.py ├── gaussVAE.jl ├── gaussVAE.py └── mixModel_grad_helper.jl ├── omniglot ├── background_filelist.txt ├── evaluation_filelist.txt ├── images_background.zip ├── images_evaluation.zip └── process_omniglot.py ├── train_DP_VAE.py ├── train_MixModel_VAE-2StochLayers.py ├── train_MixModel_VAE.jl ├── train_MixModel_VAE_vCollapsed.py ├── train_VAE.jl ├── train_reg_gaussVAE.py └── utils ├── __init__.py ├── nn_utils.jl ├── optimizers.jl └── sampling_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/LICENSE -------------------------------------------------------------------------------- /MNIST/data/process_amat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/data/process_amat.py -------------------------------------------------------------------------------- /MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt -------------------------------------------------------------------------------- /MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt.meta -------------------------------------------------------------------------------- /MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt -------------------------------------------------------------------------------- /MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/params/gaussMM_vae_params_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.ckpt.meta -------------------------------------------------------------------------------- /MNIST/params/gauss_regVae_params_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/params/gauss_regVae_params_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.ckpt -------------------------------------------------------------------------------- /MNIST/params/gauss_regVae_params_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.ckpt.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/params/gauss_regVae_params_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.ckpt.meta -------------------------------------------------------------------------------- /MNIST/params/mixVAE_params.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/params/mixVAE_params.pkl -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_-1.5--0.75-0.0-0.75-1.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component0.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component1.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component2.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component3.png -------------------------------------------------------------------------------- /MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gaussMM_vae_samples_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_component4.png -------------------------------------------------------------------------------- /MNIST/samples/gauss_regVae_samples_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/samples/gauss_regVae_samples_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.png -------------------------------------------------------------------------------- /MNIST/train_logs/dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_2.txt -------------------------------------------------------------------------------- /MNIST/train_logs/dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_3.txt -------------------------------------------------------------------------------- /MNIST/train_logs/dlgmm_trainResults_K_3_hidden_d_200_latent_d_100_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05_4.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MNIST/train_logs/dlgmm_trainResults_K_3_hidden_d_200_latent_d_20_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/dlgmm_trainResults_K_3_hidden_d_200_latent_d_20_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_1e-05.txt -------------------------------------------------------------------------------- /MNIST/train_logs/gaussMM_vae_trainResults_K_3_hidden_d_1000_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/gaussMM_vae_trainResults_K_3_hidden_d_1000_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_1.txt -------------------------------------------------------------------------------- /MNIST/train_logs/gaussMM_vae_trainResults_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/gaussMM_vae_trainResults_K_3_hidden_d_500_latent_d_25_mu_0.0-0.0-0.0_sigma_1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt -------------------------------------------------------------------------------- /MNIST/train_logs/gaussMM_vae_trainResults_K_5_hidden_d_1500_latent_d_25_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/gaussMM_vae_trainResults_K_5_hidden_d_1500_latent_d_25_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003_1.txt -------------------------------------------------------------------------------- /MNIST/train_logs/gaussMM_vae_trainResults_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/gaussMM_vae_trainResults_K_5_hidden_d_500_latent_d_5_mu_-0.5--0.25-0.0-0.25-0.5_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt -------------------------------------------------------------------------------- /MNIST/train_logs/gaussMM_vae_trainResults_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/gaussMM_vae_trainResults_K_5_hidden_d_500_latent_d_5_mu_0.0-0.0-0.0-0.0-0.0_sigma_1.0-1.0-1.0-1.0-1.0_dirichlet_alpha_1.0_adamLR_0.0003.txt -------------------------------------------------------------------------------- /MNIST/train_logs/gauss_regVae_trainResults_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/MNIST/train_logs/gauss_regVae_trainResults_hidden_d_500_latent_d_25_mu_0.0_sigma_1.0__adamLR_0.0003.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/README.md -------------------------------------------------------------------------------- /gaussMMVAE.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/gaussMMVAE.jl -------------------------------------------------------------------------------- /knn_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/knn_mnist.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gaussMMVAE.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/models/gaussMMVAE.jl -------------------------------------------------------------------------------- /models/gaussMMVAE_collapsed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/models/gaussMMVAE_collapsed.py -------------------------------------------------------------------------------- /models/gaussVAE.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/models/gaussVAE.jl -------------------------------------------------------------------------------- /models/gaussVAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/models/gaussVAE.py -------------------------------------------------------------------------------- /models/mixModel_grad_helper.jl: -------------------------------------------------------------------------------- 1 | using Distributions 2 | 3 | 4 | -------------------------------------------------------------------------------- /omniglot/background_filelist.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/omniglot/background_filelist.txt -------------------------------------------------------------------------------- /omniglot/evaluation_filelist.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/omniglot/evaluation_filelist.txt -------------------------------------------------------------------------------- /omniglot/images_background.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/omniglot/images_background.zip -------------------------------------------------------------------------------- /omniglot/images_evaluation.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/omniglot/images_evaluation.zip -------------------------------------------------------------------------------- /omniglot/process_omniglot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/omniglot/process_omniglot.py -------------------------------------------------------------------------------- /train_DP_VAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/train_DP_VAE.py -------------------------------------------------------------------------------- /train_MixModel_VAE-2StochLayers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/train_MixModel_VAE-2StochLayers.py -------------------------------------------------------------------------------- /train_MixModel_VAE.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/train_MixModel_VAE.jl -------------------------------------------------------------------------------- /train_MixModel_VAE_vCollapsed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/train_MixModel_VAE_vCollapsed.py -------------------------------------------------------------------------------- /train_VAE.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/train_VAE.jl -------------------------------------------------------------------------------- /train_reg_gaussVAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/train_reg_gaussVAE.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/nn_utils.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/utils/nn_utils.jl -------------------------------------------------------------------------------- /utils/optimizers.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/utils/optimizers.jl -------------------------------------------------------------------------------- /utils/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/enalisnick/mixture_density_VAEs/HEAD/utils/sampling_utils.py --------------------------------------------------------------------------------