gpleiss / temperature_scaling
A simple way to calibrate your neural network.
☆1,104Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for temperature_scaling
- High-quality implementations of standard and SOTA methods on a variety of tasks.☆1,452Updated last month
- The net:cal calibration framework is a Python 3 library for measuring and mitigating miscalibration of uncertainty estimates, e.g., by a …☆347Updated 3 months ago
- A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.☆551Updated 9 months ago
- Literature survey, paper reviews, experimental setups and a collection of implementations for baselines methods for predictive uncertaint…☆611Updated 2 years ago
- Bayesian active learning library for research and industrial usecases.☆869Updated 4 months ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,275Updated last month
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,593Updated 2 years ago
- ☆466Updated 3 months ago
- Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"☆454Updated last year
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆980Updated 3 months ago
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆1,437Updated this week
- PyTorch Library for Active Learning to accompany Human-in-the-Loop Machine Learning book☆943Updated last year
- A curated list of resources for Learning with Noisy Labels☆2,640Updated 6 months ago
- Pytorch Lightning code guideline for conferences☆1,242Updated last year
- This repo contains a PyTorch implementation of the paper: "Evidential Deep Learning to Quantify Classification Uncertainty"☆438Updated 10 months ago
- A learning rate range test implementation in PyTorch☆924Updated last month
- Code snippets created for the PyTorch discussion board☆545Updated 3 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆785Updated 9 months ago
- PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882☆2,005Updated last year
- mixup: Beyond Empirical Risk Minimization☆1,168Updated 3 years ago
- Corruption and Perturbation Robustness (ICLR 2019)☆1,023Updated 2 years ago
- Deep Active Learning☆811Updated 2 years ago
- A simple and effective method for detecting out-of-distribution images in neural networks.☆531Updated 3 years ago
- Deep Clustering for Unsupervised Learning of Visual Features☆1,685Updated 3 years ago
- PyTorch implementation of Contrastive Learning methods☆1,942Updated last year
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,695Updated 2 weeks ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,770Updated 9 months ago
- A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch☆1,989Updated last year
- This repository contains a collection of surveys, datasets, papers, and codes, for predictive uncertainty estimation in deep learning mo…☆576Updated last month
- GPU Accelerated t-SNE for CUDA with Python bindings☆1,802Updated last month