/BML-Project

Project code for the Bayesian Machine Learning project at ENS Paris Saclay

Primary LanguagePython

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