scikit-learn-intelex icon indicating copy to clipboard operation
scikit-learn-intelex copied to clipboard

[Enhancement] Add predict_proba support for Random Forest with `criterion="entropy"`

Open Innixma opened this issue 2 years ago • 1 comments

Version used: 2021.5.1

The following model gets a speedup with predict_proba:

from sklearnex.ensemble import RandomForestClassifier
model = RandomForestClassifier(..., criterion="gini")

However, this does not get a speedup:

from sklearnex.ensemble import RandomForestClassifier
model = RandomForestClassifier(..., criterion="entropy")

Could the entropy criterion of RandomForestClassifier be accelerated in predict_proba? When ExtraTrees support is added, this question also applies to specifying "entropy" for ExtraTrees.

Innixma avatar Mar 04 '22 00:03 Innixma

Hi, @Innixma, Thanks for request, we will analyze and plan this work, after that we will update the information

agorshk avatar Mar 04 '22 06:03 agorshk