forked from potash/lead-model
-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Description
We are currently using my branch of sklearn with the balanced feature I added to RandomForestClassifier. However, sklearn may never merge this feature. We want to be able to upgrade sklearn for bugfixes and compatibility with other packages. Two options I can think of:
-
Look into using BalancedBaggingClassifier from the imblearn package, which should be equivalent to BRF.
-
Refactor my implementation of BRF so that it can simply be an add-on to the existing sklearn codebase. Not sure how clean/easy this would be.