Support cuPy array in custom objective with the sklearn interface.
Open
feature-request
- Dominant language
- C++
- Stars
- 28.8k
- Forks
- 8.9k
- Avg merge
- 1d 12h
- Merged PRs (30d)
- 54
Description
Currently, arrays for both prediction and labels are converted into `np.ndarray`. We can avoid the data transfer by:
- Output GPU prediction in the sklearn interface.
- Return GPU array in `DMatrix.get_float_info`
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.