Skip to main content

Instance-Based Uncertainty Estimation for Gradient-Boosted Regression Trees

Project description

IBUG: Instance-Based Uncertainty Estimation for Gradient-Boosted Regression Trees

PyPi version Python version Github License Build

ibug is a simple wrapper that extends any gradient-boosted regression trees (GBRT) model into a probabilistic estimator, and is compatible with all major GBRT frameworks including LightGBM, XGBoost, CatBoost, and SKLearn.

Install

pip install ibug

Experiment

Download a Dataset

  1. Follow the instructions in the readme for a dataset in data/.

Quantify Prediction Uncertainy with LightGBM + IBUG

python3 scripts/experiments/train.py --tree_type lgb
python3 scripts/experiments/predict.py --tree_type lgb

License

Apache License 2.0.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ibug-0.0.3.tar.gz (279.0 kB view hashes)

Uploaded Source

Built Distributions

ibug-0.0.3-cp310-cp310-win_amd64.whl (436.1 kB view hashes)

Uploaded CPython 3.10 Windows x86-64

ibug-0.0.3-cp310-cp310-win32.whl (415.6 kB view hashes)

Uploaded CPython 3.10 Windows x86

ibug-0.0.3-cp310-cp310-musllinux_1_1_x86_64.whl (1.3 MB view hashes)

Uploaded CPython 3.10 musllinux: musl 1.1+ x86-64

ibug-0.0.3-cp310-cp310-musllinux_1_1_i686.whl (1.2 MB view hashes)

Uploaded CPython 3.10 musllinux: musl 1.1+ i686

ibug-0.0.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.3 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ x86-64

ibug-0.0.3-cp310-cp310-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl (1.2 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ i686 manylinux: glibc 2.5+ i686

ibug-0.0.3-cp310-cp310-macosx_10_9_x86_64.whl (468.4 kB view hashes)

Uploaded CPython 3.10 macOS 10.9+ x86-64

ibug-0.0.3-cp39-cp39-win_amd64.whl (435.8 kB view hashes)

Uploaded CPython 3.9 Windows x86-64

ibug-0.0.3-cp39-cp39-win32.whl (415.2 kB view hashes)

Uploaded CPython 3.9 Windows x86

ibug-0.0.3-cp39-cp39-musllinux_1_1_x86_64.whl (1.3 MB view hashes)

Uploaded CPython 3.9 musllinux: musl 1.1+ x86-64

ibug-0.0.3-cp39-cp39-musllinux_1_1_i686.whl (1.2 MB view hashes)

Uploaded CPython 3.9 musllinux: musl 1.1+ i686

ibug-0.0.3-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.3 MB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.17+ x86-64

ibug-0.0.3-cp39-cp39-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl (1.2 MB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.17+ i686 manylinux: glibc 2.5+ i686

ibug-0.0.3-cp39-cp39-macosx_10_9_x86_64.whl (468.4 kB view hashes)

Uploaded CPython 3.9 macOS 10.9+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page