ryokamoi / sg_langevin

Implementation of Stochastic Gradient Langevin Dynamics

Implementation of "Bayesian Learning via Stochastic Gradient Langevin Dynamics"

This is not the original implementation.

Bayesian Learning via Stochastic Gradient Langevin Dynamics

Environment

• Python 3.6.2
• TensorFlow 1.10
• See pip.freeze for details

Usage

Logistic regression

python main.py --hparams params/SGLD_LR

Results

Logistic regression

Author

Ryo Kamoi