Skip to content

Latest commit

 

History

History
35 lines (25 loc) · 1.19 KB

README.md

File metadata and controls

35 lines (25 loc) · 1.19 KB

Dropout as Bayesian Approximation for Deep Neural Network

Open In Colab

You can run the whole notebook in Colab without issue. Getting helper scripts from Github repo, downloading datasets, and install required packages are all handled.

Running on local machine

  • Clone repository and change directory to local copy
  • Create a conda environment conda create --name bayesian_nn
  • Activate conda environment conda activate bayesian_nn
  • Install required libraries pip install -r requirements.txt
  • Run jupyter lab jupyter lab
  • Open Dropout as Bayesian Approximation.ipynb

Visualizing Model Uncertainty

Data is passed through the model with Dropout turned on during the inference step for multiple times. The resulting probability distributions are what the model produce over multiple passes.

When the model is uncertain: Uncertain

When the model is certain: Certain

High certainty: High_Certain

Extremely high certainty: Ext_Certain