chimeraboost 0.7.1

CatBoost-inspired gradient boosting in pure Python with a numba backend

chimeraboost

What if CatBoost, but 30x faster, slightly worse, and all in Python?

⚠️ Project is in active development: breaking changes should be expected.

• Installation

pip install chimeraboost

• Sample code:

from chimeraboost import ChimeraBoostClassifier, ChimeraBoostRegressor

# classification
clf = ChimeraBoostClassifier(early_stopping=True)
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)
reg.fit(X, y)

• What?

  • GBDT library that only depends on common Python libraries
  • Supports regression, quantile regression, binary and multiclass classification.
  • Categorical features, sample weights, and automatic early stopping
  • Within ~3% F1 / ~5% RMSE of CatBoost on a 34-dataset OpenML benchmark, at ~30× the speed

• Why?

  • I want to be able to modify my GBDT library at will
  • I know Python and I don't know C