bayesian-neural-network-mnist icon indicating copy to clipboard operation
bayesian-neural-network-mnist copied to clipboard

Bayesian neural network using Pyro and PyTorch on MNIST dataset

Bayesian neural network using Pyro and PyTorch on MNIST dataset

Jupyter notebook corresponding to tutorial: Getting your Neural Network to Say "I Don't Know" - Bayesian NNs using Pyro and Pytorch

Requires following packages:

  • PyTorch
  • Pyro
  • Numpy
  • Matplotlib

Made by @paraschopra

MIT License.