boosters#

boosters is a high-performance gradient boosting library for Python and Rust.

โšก Fast ยท Rust core | ๐Ÿ”Œ Compatible ยท sklearn API | ๐ŸŽฏ Flexible ยท GBDT & GBLinear | ๐Ÿ” Explainable ยท SHAP values |

Installation#

pip install boosters
uv pip install boosters
git clone https://github.com/egordm/booste-rs
cd booste-rs
pip install -e packages/boosters-python

Quick Example#

from boosters.sklearn import GBDTRegressor

model = GBDTRegressor(n_estimators=100, max_depth=6)
model.fit(X_train, y_train)
predictions = model.predict(X_test)

Learn#

How to use boosters?
Hands-on examples

Reference#

API Documentation
Research & Design

Community#

Contributing
Tools

Indices and Tables#