simple-uq

Uncertainty quantification models implemented simply and cleanly using pytorch-lightning.

---

UNDER CONSTRUCTION

Models

• Probabilistic Neural Net (PNN): A neural network that outputs the mean and variance of a Normal distribution.

• More coming soon!

Installation Instructions

git clone https://github.com/uncertainty-toolbox/simple-uq
cd simple-uq
pip install -e .

Using the Library

For example usage of the models, see the examples directory. More information and tips on using pytorch-lightning can be found in their tutorials and documentation.