Closed
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.train
.
Because train()
expects to be given a LightGBM Dataset
object, this also implies creating a new class lightgbm.dask.DaskDataset
. train()
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.train()
.
References
See the DaskDMatrix
in xgboost.dask
for some inspiration on how DaskDataset
might be implemented.