Support GPU-backed data for metrics. - Githubissues

dask / dask-ml

Scalable Machine Learning with Dask

http://ml.dask.org

BSD 3-Clause "New" or "Revised" License

905 stars 256 forks source link

Support GPU-backed data for metrics. #991

Open trivialfis opened 7 months ago

trivialfis commented 7 months ago

Currently, the dask-ml assumes CPU-backed data input with sklearn and np operations like https://github.com/dask/dask-ml/blob/b95ba909c6dcd37c566f5193ba0b918396edaaee/dask_ml/metrics/classification.py#L105