christophM / rulefit

Python implementation of the rulefit algorithm
MIT License
411 stars 112 forks source link

Add get_feature_importance method to RuleFit class #41

Closed caseywhorton closed 3 years ago

caseywhorton commented 3 years ago

Hello! I've taken much interest in this topic and would like to contribute to this repo.

This method uses the rule set from the get_rules method in the RuleFit class and the submitted features to find the importance of the features either globally or over a subregion.

chriswbartley commented 3 years ago

Thanks Casey! Appreciate the input 😊