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

[ENH] Add GPU Tests and Docs for Hyperparameter Tuning, Automl , Exporting Trained Model and end to end examples #397

Open
VibhuJawa opened this issue Feb 9, 2022 · 2 comments
Labels
enhancement New feature or request machine learning Improvements or issues with machine learning functionality

Comments

@VibhuJawa
Copy link
Collaborator

Is your feature request related to a problem? Please describe.

In #394 we are adding GPU section for ML model training.

All other sections (see below) of the machine-learning docs, they should probably work with GPU's . We don't have have tests for those now though. I think it might be worth-while adding tests for them and then adding the GPU sections for those too.

Hyperparameter Tuning
Auto ML
Exporting Trained Model
End to End Example

CC: @charlesbluca

@VibhuJawa VibhuJawa added enhancement New feature or request needs triage Awaiting triage by a dask-sql maintainer machine learning Improvements or issues with machine learning functionality and removed needs triage Awaiting triage by a dask-sql maintainer labels Feb 9, 2022
@sarahyurick
Copy link
Collaborator

I think this would also close #130

@sarahyurick
Copy link
Collaborator

I plan to pick this up as I investigate #943 and #1020

I'll also be sure to note parameters that are helpful when doing GPU ML, like gpu_hist for XGBoost.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request machine learning Improvements or issues with machine learning functionality
Projects
None yet
Development

No branches or pull requests

2 participants