amjams / FeatBoost

Boosted Iterative Input Selection
20 stars 6 forks source link

FeatBoost

Python implementation of FeatBoost. See the paper for details.

Demo

This demo compares featboost to feature ranking of XGBoost on the Madelon benchmark dataset. Open In Colab

Usage

pip install git+https://github.com/amjams/FeatBoost.git

Or just clone the repo (recommended for now)

git clone https://github.com/amjams/FeatBoost.git

Basic usage of the model (see the demo above for more details).

from featboost import FeatBoostClassifier

clf = FeatBoostClassifier()
clf.fit(X, y)
clf.feature_importances_