Uncertainty Estimation Python Packages

ngboost

Natural Gradient Boosting for Probabilistic Prediction

178K 2K 249

quantile-forest

Quantile Regression Forests compatible with scikit-learn.

60K 254 30

uncertainty-toolbox

Uncertainty Toolbox: a Python toolbox for predictive uncertainty quantification, calibration, metrics, and visualization

13K 2K 146

puncc

👋 Puncc is a python library for predictive uncertainty quantification using conformal prediction.

6K 391 17

uqlm

UQLM: Uncertainty Quantification for Language Models, is a Python package for UQ-based LLM hallucination detection

5K 1K 123

pygpc

A Sensitivity and uncertainty analysis toolbox for Python based on the generalized polynomial chaos method

4K 85 28

xgboostlss

An extension of XGBoost to probabilistic modelling

4K 709 80

lightgbmlss

An extension of LightGBM to probabilistic modelling

3K 384 36

bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch

2K 10 3

adfwi-torch

An Automatic Differentiation-based Waveform Inversion Framework Implemented in PyTorch.

2K 190 27

nlp-uncertainty-zoo

Model zoo for different kinds of uncertainty quantification methods used in Natural Language Processing, implemented in PyTorch.

1K 55 4

torchbayesian

PyTorch-native Bayesian neural networks for uncertainty quantification in deep learning

440 6 1

photonai-conformal

No description available

295 1 0

robust-deep-learning

A project to train your model from scratch or fine-tune a pretrained model using the losses provided in this library to improve out-of-distribution detection and uncertainty estimation performances. Calibrate your model to produce enhanced uncertainty estimations. Detect out-of-distribution data using the defined score type and threshold.

255 18 3