Skip to content
This repository has been archived by the owner on Jan 14, 2022. It is now read-only.
/ rulefit Public archive

Python implementation of rulefit algorithm with support for xgboost.

Notifications You must be signed in to change notification settings

bishax/rulefit

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

NOTE: This code is incomplete, and untested.

Python implementation of the rulefit (http://statweb.stanford.edu/~jhf/ftp/RuleFit.pdf) algorithm (with support for xgboost).

The algorithm is a multi-step process:

  1. Generate a tree ensemble using random forest/gradient boosting
  2. Use the trees to form rules, with each decision path in a tree forming one rule.
  3. Prune the rules and the original input features using L1-regularised regression (LASSO)

Largely written before discovering the more complete implementation here: https://github.com/christophM/rulefit

About

Python implementation of rulefit algorithm with support for xgboost.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages