bayesian-neural-network-mnist
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.