Google Git
Sign in
apache / mxnet-test / refs/tags/v0.10.11 / . / example / bayesian-methods
tree: e9f83c1862e47d774308731bff1601c189cbc660 [path history] [tgz]
  1. algos.py
  2. bdk.ipynb
  3. bdk_demo.py
  4. data_loader.py
  5. README.md
  6. sgld.ipynb
  7. utils.py
example/bayesian-methods/README.md

Bayesian Methods

This folder contains examples related to Bayesian Methods.

We currently have Stochastic Gradient Langevin Dynamics (SGLD) (Welling and Teh, 2011) and Bayesian Dark Knowledge (BDK) (Balan, Rathod, Murphy and Welling, 2015).

sgld.ipynb shows how to use MXNet to repeat the toy experiment in the original SGLD paper.

bdk.ipynb shows how to use MXNet to implement the DistilledSGLD algorithm in Bayesian Dark Knowledge.

bdk_demo.py contains scripts (more than the notebook) related to Bayesian Dark Knowledge. Use python bdk_demo.py -d 1 -l 2 -t 50000 to run classification on MNIST.