search

christophM / rulefit

Python implementation of the rulefit algorithm
MIT License
406 stars 111 forks source link

Add Probability Prediction Method #37

Closed ajshearer124 closed 4 years ago

ajshearer124 commented 4 years ago

Add wrapper method to predict probabilities via predict_proba, if that method exists in the sklearn class lscv.

ajshearer124 commented 4 years ago

@christophM think this is a pretty straightforward addition, would like to see it merged in!

christophM commented 4 years ago

Hey, I just saw this. Thanks for opening this pull request! Could you add a test as well? Then I think it would be ready to merge.

ajshearer124 commented 4 years ago

Of course - added a separate script for some simple testing. Lmk what you think!

christophM commented 4 years ago

Thanks!