Support GPU-backed data for metrics.
Open
- Dominant language
- Python
- Stars
- 951
- Forks
- 262
- PR merge metrics
- No merged PRs in 30d
Description
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
Contributor guide
Assessment
This issue has not been assessed yet.