You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
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.
Summary
As of this writing,
lightgbm.dask
only supports model classes that mimic the scikit-learn API. It should also support a function, equivalent tolightgbm.engine.cv
.LightGBM/python-package/lightgbm/engine.py
Line 391 in 36322ce
Because
cv()
expects to be given a LightGBMDataset
object, this also implies creating a new classlightgbm.dask.DaskDataset
.cv()
should take intrain_set
as alightgbm.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
inxgboost.dask
for some inspiration on howDaskDataset
might be implemented.https://github.com/dmlc/xgboost/blob/a275f4026728ed14fbc70da142ef7a4a1d3de04d/python-package/xgboost/dask.py#L181-L186
DaskDataset
may be implemented outside of this feature, to support #3846The text was updated successfully, but these errors were encountered: