A simple way to calibrate your neural network.
☆1,168Jul 26, 2025Updated 7 months ago
Alternatives and similar repositories for temperature_scaling
Users that are interested in temperature_scaling are comparing it to the libraries listed below
Sorting:
- Calibration of Convolutional Neural Networks☆170Apr 19, 2023Updated 2 years ago
- The net:cal calibration framework is a Python 3 library for measuring and mitigating miscalibration of uncertainty estimates, e.g., by a …☆373Jan 14, 2026Updated last month
- Code for the paper "Calibrating Deep Neural Networks using Focal Loss"☆160Jan 10, 2024Updated 2 years ago
- High-quality implementations of standard and SOTA methods on a variety of tasks.☆1,567Feb 2, 2026Updated last month
- Reliability diagrams visualize whether a classifier model needs calibration☆167Feb 11, 2022Updated 4 years ago
- Calibration library and code for the paper: Verified Uncertainty Calibration. Ananya Kumar, Percy Liang, Tengyu Ma. NeurIPS 2019 (Spotlig…☆153Nov 10, 2022Updated 3 years ago
- NumPy library for calibration metrics☆75Feb 24, 2026Updated last week
- Local Temperature Scaling for Probability Calibration☆22Nov 26, 2021Updated 4 years ago
- A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networks☆233Dec 27, 2018Updated 7 years ago
- Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"☆478Jul 6, 2023Updated 2 years ago
- ☆38Nov 13, 2020Updated 5 years ago
- A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.☆592Jan 26, 2024Updated 2 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆988May 30, 2025Updated 9 months ago
- Reliability diagrams, Platt's scaling, isotonic regression☆75Aug 2, 2014Updated 11 years ago
- Deep Anomaly Detection with Outlier Exposure (ICLR 2019)☆575Oct 9, 2021Updated 4 years ago
- This repo contains a PyTorch implementation of the paper: "Evidential Deep Learning to Quantify Classification Uncertainty"☆513Jan 2, 2024Updated 2 years ago
- Code for "Uncertainty Estimation Using a Single Deep Deterministic Neural Network"☆275Mar 17, 2022Updated 3 years ago
- Training Confidence-Calibrated Classifier for Detecting Out-of-Distribution Samples / ICLR 2018☆182Apr 8, 2020Updated 5 years ago
- Code implementation of our ICLR'21 paper "Calibration of Neural Networks using Splines"☆22Mar 31, 2023Updated 2 years ago
- A simple and effective method for detecting out-of-distribution images in neural networks.☆542Oct 12, 2021Updated 4 years ago
- ☆472Feb 2, 2026Updated last month
- Adaptive-binning for evaluation of confidence calibration☆12Jul 28, 2019Updated 6 years ago
- Literature survey, paper reviews, experimental setups and a collection of implementations for baselines methods for predictive uncertaint…☆639Aug 1, 2022Updated 3 years ago
- Principled Detection of Out-of-Distribution Examples in Neural Networks☆202Jun 28, 2017Updated 8 years ago
- Code for "Well-calibrated Model Uncertainty with Temperature Scaling for Dropout Variational Inference" (NeurIPS Bayesian Deep Learning W…☆24Jan 25, 2020Updated 6 years ago
- Uncertainty Toolbox: a Python toolbox for predictive uncertainty quantification, calibration, metrics, and visualization☆1,977Mar 5, 2025Updated last year
- Code and datasets for the EMNLP 2020 paper "Calibration of Pre-trained Transformers"☆60Jun 12, 2023Updated 2 years ago
- ☆21Mar 25, 2023Updated 2 years ago
- ☆428Aug 28, 2021Updated 4 years ago
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,838Updated this week
- Code for the paper "A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attacks".☆351Aug 13, 2019Updated 6 years ago
- Bayesian Deep Learning Benchmarks☆672Mar 24, 2023Updated 2 years ago
- Corruption and Perturbation Robustness (ICLR 2019)☆1,138Aug 24, 2022Updated 3 years ago
- Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more☆1,957Oct 20, 2023Updated 2 years ago
- Wrapper for a PyTorch classifier which allows it to output prediction sets. The sets are theoretically guaranteed to contain the true cla…☆255Jan 23, 2023Updated 3 years ago
- A scikit-learn compatible neural network library that wraps PyTorch☆6,151Feb 25, 2026Updated last week
- Model interpretability and understanding for PyTorch☆5,564Feb 26, 2026Updated last week
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,612Jun 16, 2021Updated 4 years ago
- Non-Parametric Calibration for Classification (AISTATS 2020)☆18Feb 3, 2022Updated 4 years ago