gpleiss / temperature_scaling
A simple way to calibrate your neural network.
☆1,135Updated 3 years ago
Alternatives and similar repositories for temperature_scaling:
Users that are interested in temperature_scaling are comparing it to the libraries listed below
- High-quality implementations of standard and SOTA methods on a variety of tasks.☆1,497Updated last month
- The net:cal calibration framework is a Python 3 library for measuring and mitigating miscalibration of uncertainty estimates, e.g., by a …☆356Updated 7 months ago
- Literature survey, paper reviews, experimental setups and a collection of implementations for baselines methods for predictive uncertaint…☆618Updated 2 years ago
- ☆468Updated 8 months ago
- A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.☆561Updated last year
- Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"☆458Updated last year
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,294Updated 2 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,610Updated 3 years ago
- PyTorch Boilerplate For Research☆609Updated 3 years ago
- A Survey☆552Updated 2 years ago
- Bayesian active learning library for research and industrial usecases.☆894Updated 9 months ago
- PyTorch Library for Active Learning to accompany Human-in-the-Loop Machine Learning book☆960Updated 2 years ago
- Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learning, ICLR 2020☆235Updated 2 years ago
- PyTorch implementation of Contrastive Learning methods☆1,969Updated last year
- Corruption and Perturbation Robustness (ICLR 2019)☆1,060Updated 2 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆984Updated last week
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆1,472Updated this week
- torch-optimizer -- collection of optimizers for Pytorch☆3,096Updated last year
- A curated list of resources for Learning with Noisy Labels☆2,666Updated 10 months ago
- A learning rate range test implementation in PyTorch☆951Updated 4 months ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,718Updated 3 weeks ago
- Calibration library and code for the paper: Verified Uncertainty Calibration. Ananya Kumar, Percy Liang, Tengyu Ma. NeurIPS 2019 (Spotlig…☆147Updated 2 years ago
- Calibration of Convolutional Neural Networks☆161Updated last year
- Code snippets created for the PyTorch discussion board☆561Updated 4 years ago
- mixup: Beyond Empirical Risk Minimization☆1,178Updated 3 years ago
- Stochastic Weight Averaging in PyTorch☆969Updated 3 years ago
- Code for the paper "Calibrating Deep Neural Networks using Focal Loss"☆160Updated last year
- Understanding Deep Networks via Extremal Perturbations and Smooth Masks☆345Updated 4 years ago
- Reliability diagrams visualize whether a classifier model needs calibration☆148Updated 3 years ago
- Fast, differentiable sorting and ranking in PyTorch☆799Updated last year