/sg_langevin

Implementation of Stochastic Gradient Langevin Dynamics

Primary LanguagePythonMIT LicenseMIT

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