Bayesian Machine Learning Project
In this project, we implemented and trained a Neural Network using the Stochastic Gradient Langevin Dynamics.
How to run
In order to visualize some results the user can run the model.py
file.
Note that the two interesting variables to play with are the batch numbers and the sequence of learning rates.
The model is based on the code provided by Javier Antoran at SGLD