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
To install, run:
pip install .
Examples for usage can be found in the scripts folder.