Skip to content

leetim13/Variational-Autoencoder-on-MNIST

Repository files navigation

This project implements and investigates the Variational Autoencoder on binarized MNIST digits by building a generative model to infer the bottom half of the given binarized MNIST digits conditioned on the top half of these images.

Implementations include:

  • Project.toml packages for the Julia environment.
  • variational_autoencoder.py Python version.
  • loadMNIST.py loading MNIST data in Python.
  • example_flux_model.jl example flux model in Julia.
  • vae.jl source code in Julia.
  • encoder_params.bson final params/weights of trained model.
  • decoder_params.bson final params/weights of trained model.
  • Julia-Variational-Autoencoder-Final.ipynb the final jupyter notebook project.

Note: this project is part of the assignment from Statistical Methods for Machine Learning II at the Univeristy of Toronto.

About

This project implements and investigates the Variational Autoencoder on binarized MNIST digits by building a generative model to infer the bottom half of the given binarized MNIST digits conditioned on the top half of these images.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors