Uncertainty Networks

Implementation of MC Dropout and Deep Ensembles for model uncertainty estimation in deep neural networks.

Features:

• Estimation of model (epistemic) uncertainty for MLPs and GRUs
• Supports propagation of uncertainty across different modules
• Example script to train network that estimate both model & data uncertainties

Installation

To install, run:

pip install .

Usage

Examples for usage can be found in the scripts folder.