random-forest-importances icon indicating copy to clipboard operation
random-forest-importances copied to clipboard

Published 20 hours ago • verified publisher icon parrt

Add multiprocessing for oob_importances

Open RohanBhandari opened this issue 5 years ago • 1 comments

Wanted to suggest parallelizing the oob importance calculation in order to speed it up, since the importances can be calculated independently for each feature.

In my use case, I saw a >8x reduction in runtime after parallelizing and would be happy to contribute the code needed to implement this.

RohanBhandari avatar Mar 20 '19 00:03 RohanBhandari

I'd LOVE to see a pull request for that or even sample code. I barely figured out how to get OOB samples from sklearn ;)

parrt avatar Mar 20 '19 00:03 parrt