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

Support GPU acceleration? #8

Open
hhaootian opened this issue Sep 6, 2022 · 2 comments
Open

Support GPU acceleration? #8

hhaootian opened this issue Sep 6, 2022 · 2 comments

Comments

@hhaootian
Copy link

Describe the issue

Does LCE support GPU acceleration? One example in XGBoost is to set tree_method as gpu_hist, but I did not find this param in LCE.

Explain the improvement

No response

@LocalCascadeEnsemble
Copy link
Owner

Hello,

Thank you for your message.

No, LCE current implementation does not support GPU.
LCE depends on scikit-learn, which does not support GPU as of today (http://scikit-learn.org/stable/faq.html#will-you-add-gpu-support).
I will follow the development on this aspect.

Best,

@Victordmz
Copy link

Victordmz commented Apr 16, 2023

tree_method='gpu_hist' is very easy to add to all the XGBoost initializations in _xgboost.py, for instance here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants