fit_lightgbm_es: Fit GBDT by lightgbm package with early stopping

In five-dots/ml4e: Machine Learning for Everyone

Description

Fit GBDT by lightgbm package with early stopping

Usage

fit_lightgbm_es(data, eval, task_type, ...)

Arguments

data	A training data of lightgbm::lgb.Dataset class object.
eval	A evaluation data of lightgbm::lgb.Dataset class object. Used for early stopping.
task_type	A character scalar of task_type.
...	Additional arguments passed to lightgbm::lgb.train].

Value

A fitted object.

five-dots/ml4e documentation built on June 19, 2020, 4:26 p.m.