Skip to content

Make some functions from Dask module class methods #3832

@StrikerRUS

Description

@StrikerRUS

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.

def _train(client, data, label, params, model_factory, sample_weight=None, group=None, **kwargs):

def _predict(model, data, raw_score=False, pred_proba=False, pred_leaf=False, pred_contrib=False,

#3515 (comment)

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions