Description
Summary
As of this writing, lightgbm.dask
only supports model classes that mimic the scikit-learn API. It should also support a function, equivalent to lightgbm.engine.cv
.
LightGBM/python-package/lightgbm/engine.py
Line 391 in 36322ce
Because cv()
expects to be given a LightGBM Dataset
object, this also implies creating a new class lightgbm.dask.DaskDataset
. cv()
should take in train_set
as a lightgbm.dask.DaskDataset
, and should return a regular LightGBM Booster.
Motivation
Having a functional interface would make the transition from non-Dask to Dask easier for users who are already using lightgbm.engine.cv)
.
References
See the DaskDMatrix
in xgboost.dask
for some inspiration on how DaskDataset
might be implemented.
DaskDataset
may be implemented outside of this feature, to support #3846