dask-sql icon indicating copy to clipboard operation
dask-sql copied to clipboard

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

Open VibhuJawa opened this issue 3 years ago • 2 comments

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

In https://github.com/dask-contrib/dask-sql/pull/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 avatar Feb 09 '22 17:02 VibhuJawa

I think this would also close https://github.com/dask-contrib/dask-sql/issues/130

sarahyurick avatar Jan 31 '23 18:01 sarahyurick

I plan to pick this up as I investigate https://github.com/dask-contrib/dask-sql/issues/943 and https://github.com/dask-contrib/dask-sql/issues/1020

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

sarahyurick avatar Feb 13 '23 18:02 sarahyurick