CatBoost-inspired gradient boosting in pure Python with a numba backend
Project description
chimeraboost
What if CatBoost was slightly worse, 4× faster, and all in Python?
📖 Documentation: bbstats.github.io/chimeraboost
- Installation
pip install chimeraboost
- Sample code:
from chimeraboost import ChimeraBoostClassifier, ChimeraBoostRegressor
# classification
clf = ChimeraBoostClassifier(early_stopping=True, n_ensembles=2)
clf.fit(X, y, cat_features=[0, 1], sample_weight=w)
proba = clf.predict_proba(X_test)
# regression (RMSE, MAE, or Quantile)
reg = ChimeraBoostRegressor(loss="Quantile", alpha=0.9, early_stopping=True, n_ensembles=10)
reg.fit(X, y)
-
What?
- Exceedingly opinionated GBDT library that only depends on common Python libraries
- Categorical features (catboost-like processing) and sample weights
- Bagging as a first-class feature (
n_ensembles) - Automatic early stopping, with optional grouped splitting for the validation set
- Supports regression, quantile regression, binary and multiclass classification
- Exact SHAP explanations (
model.shap_values(X)) — interventional TreeSHAP computed exactly thanks to the oblivious tree structure
- Exceedingly opinionated GBDT library that only depends on common Python libraries
-
Tuning tips
- Interaction-heavy regression: raise
depthto 8–10 (default 6 is conservative to protect small data).
- Interaction-heavy regression: raise
-
Inspirations / Citations
- CatBoost — Prokhorenkova et al., NeurIPS 2018 — ordered boosting, ordered target statistics, oblivious trees
- XGBoost — Chen & Guestrin, KDD 2016 — regularized objective, Newton leaf estimation, column subsampling
- LightGBM — Ke et al., NeurIPS 2017 — histogram-based split finding
- Linear-leaf trees — Shi et al., IJCAI 2019 (arXiv:1802.05640) — piece-wise-linear regression trees (
linear_leaves) - TreeSHAP — Lundberg et al., Nature Machine Intelligence 2020 (orig. SHAP, NeurIPS 2017) — exact additive feature attributions (
shap_values) - Hierarchical shrinkage — Agarwal et al., ICML 2022 (arXiv:2202.00858) — leaf regularization (
hs_lambda) - TabArena — Erickson et al., NeurIPS 2025 (arXiv:2506.16791) — tabular benchmark used for evaluation
-
Why?
- I want to be able to modify my GBDT library at will
- I know Python and I don't know C
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
chimeraboost-0.14.2.tar.gz
(78.3 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file chimeraboost-0.14.2.tar.gz.
File metadata
- Download URL: chimeraboost-0.14.2.tar.gz
- Upload date:
- Size: 78.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.13.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
965bc5a71b1c37d31784912911ecf103262277c8ef7a93ffb4b89ba838ec5bfe
|
|
| MD5 |
ca4f561f2e3176c69d31c0e0dfae4f37
|
|
| BLAKE2b-256 |
e09027dc7e2dc2f931e7d9193966c23eb45a519f3117fa7f1b155fafd0c05864
|
File details
Details for the file chimeraboost-0.14.2-py3-none-any.whl.
File metadata
- Download URL: chimeraboost-0.14.2-py3-none-any.whl
- Upload date:
- Size: 58.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.13.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
251da296d8446b9de9439cca5c011f473e3ea8fae613f94b1f881f1a9947d1b8
|
|
| MD5 |
3a38f40ff88564ee5b83c132f2140d85
|
|
| BLAKE2b-256 |
be70581a8ec7808133eba5d72d45781cb44f98fae28d3c93e91d30cac8366eb0
|