-
-
Notifications
You must be signed in to change notification settings - Fork 26.5k
Closed
Labels
EasyWell-defined and straightforward way to resolveWell-defined and straightforward way to resolveNew Featuremodule:ensemble
Milestone
Description
Describe the workflow you want to enable
I want to officially use the Poisson splitting criterion in RandomForestRegressor.
Describe your proposed solution
#17386 implemented the poisson splitting criterion for DecisionTreeRegressor and ExtraTreeRegressor. This also enabled—somewhat silently—to do:
import numpy as np
from sklearn.ensemble import RandomForestRegressor
y = [0, 1, 2]
X = np.arange(6).reshape(3, 2)
rf = RandomForestRegressor(criterion="poisson")
rf.fit(X, y)
Note: The same is true for ensemble.ExtraTreesRegressor.
Tasks:
- Add the poisson splitting criterion to the docstring of
RandomForestRegressor. - Add input validation (non-negative
y) toRandomForestRegressor. - Expand the tests for
RandomForestRegressor.
Metadata
Metadata
Assignees
Labels
EasyWell-defined and straightforward way to resolveWell-defined and straightforward way to resolveNew Featuremodule:ensemble