Skip to content

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.

Notifications You must be signed in to change notification settings

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

No packages published