Skip to content

thalassemia/rulefit

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

rulefit

Full Disclaimer: This is based heavily on the original Python RuleFit implementation by Christopher Molnar (check out his Github repo).

The main tweaks that I've made are listed below:

  1. Uses LightGBM instead of scikit-learn's built-in tree ensemble functions
  2. Uses scikit-learn's stochastic gradient descent functions (SGDRegressor/SGDClassifier) instead of LassoCV/LogisticRegressionCV

In my limited testing, the first tweak significantly improves the achievable accuracy while the second drastically reduces training time.

Note that the default parameters are tuned for my specific use case and will likely need to be modified for other data.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages