Python code for Dirichlet calibration
Project description
Dirichlet Calibration Python implementation
This is a Python implementation of the Dirichlet Calibration presented in Beyond temperature scaling: Obtaining well-calibrated multi-class probabilities with Dirichlet calibration at NeurIPS 2019.
Installation
# Clone the repository
git clone git@github.com:dirichletcal/dirichlet_python.git
# Go into the folder
cd dirichlet_python
# Create a new virtual environment with Python3
python3.6 -m venv venv
# Load the generated virtual environment
source venv/bin/activate
# Upgrade pip
pip install --upgrade pip
# Install all the dependencies
pip install -r requirements.txt
Unittest
python -m unittest discover dirichletcal
Cite
If you use this code in a publication please cite the following paper
@inproceedings{kull2019dircal,
title={Beyond temperature scaling: Obtaining well-calibrated multi-class probabilities with Dirichlet calibration},
author={Kull, Meelis and Nieto, Miquel Perello and K{\"a}ngsepp, Markus and Silva Filho, Telmo and Song, Hao and Flach, Peter},
booktitle={Advances in Neural Information Processing Systems},
pages={12295--12305},
year={2019}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
dirichletcal-0.1.dev1.tar.gz
(12.6 kB
view hashes)
Built Distribution
Close
Hashes for dirichletcal-0.1.dev1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1a6d4258e6b1d7bba752c5d60fab9843489c892fbd2eec955b357e5fcb9cbef8 |
|
MD5 | 630620d7cd1111fab48ab53f82dc2dcf |
|
BLAKE2b-256 | e63d11d4e93c5b5d6cbdd1ed453a656ce9ac9481b7999edaf0ff4c14572aa52f |