Uncertainty quantification models implemented simply and cleanly using pytorch-lightning.
-
Probabilistic Neural Net (PNN): A neural network that outputs the mean and variance of a Normal distribution.
-
More coming soon!
git clone https://github.com/uncertainty-toolbox/simple-uq
cd simple-uq
pip install -e .
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.