-
Notifications
You must be signed in to change notification settings - Fork 4k
Closed
Labels
Description
It will be good to have _train() and _predict() functions (at least them, very possibly some other functions can be made methods) in a form of _LGBMModel's methods. It will improve encapsulation in Dask module.
LightGBM/python-package/lightgbm/dask.py
Line 179 in ac706e1
| def _train(client, data, label, params, model_factory, sample_weight=None, group=None, **kwargs): |
LightGBM/python-package/lightgbm/dask.py
Line 337 in ac706e1
| def _predict(model, data, raw_score=False, pred_proba=False, pred_leaf=False, pred_contrib=False, |