Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEA] Obtaining individual tree results of a random forest #6115

Open
YuelingMa0 opened this issue Oct 18, 2024 · 0 comments
Open

[FEA] Obtaining individual tree results of a random forest #6115

YuelingMa0 opened this issue Oct 18, 2024 · 0 comments
Labels
? - Needs Triage Need team to review and classify feature request New feature or request

Comments

@YuelingMa0
Copy link

Is your feature request related to a problem? Please describe.
I wish I could use cuml to obtain individual tree results of a random forest. However, this function is not supported in the current cuml package. Using the random forest regression function in the current cuml package, I can only obtain the average of tree results.

Describe the solution you'd like
An attribute in the existing random forest regression function to provide results from each tree.

Describe alternatives you've considered
I have been using the "estimator_" in the RandomForestRegressor function of scikit-learn to obtain individual tree ouputs, but that package only works on CPUs.

Additional context
https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html

@YuelingMa0 YuelingMa0 added ? - Needs Triage Need team to review and classify feature request New feature or request labels Oct 18, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
? - Needs Triage Need team to review and classify feature request New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant