Bayesian CNN Digit Classifier

An app for drawing digits which are classified with a bayesian neural network classifier.

Bayesian NN's provide us with a few nice benefits:

• Captures uncertainty
• Inherent regularisation

A forward pass through a normal neural network is deterministic and using the parameters configured during training. Whereas Bayesian Neural Networks are non-deterministic as they swap weights for probability distributions which you sample from in a forward pass.

Try the app here.

Run the app locally

Run streamlit run bayesian-digit-classifier/app.py.

Re-train the model

Run python bayesian-digit-classifier/training.py.

Demo