lightgbm.early_stopping¶
-
lightgbm.
early_stopping
(stopping_rounds, first_metric_only=False, verbose=True)[source]¶ Create a callback that activates early stopping.
Activates early stopping. The model will train until the validation score stops improving. Validation score needs to improve at least every
early_stopping_rounds
round(s) to continue training. Requires at least one validation data and one metric. If there’s more than one, will check all of them. But the training data is ignored anyway. To check only the first metric setfirst_metric_only
to True.- Parameters
stopping_rounds (int) – The possible number of rounds without the trend occurrence.
first_metric_only (bool, optional (default=False)) – Whether to use only the first metric for early stopping.
verbose (bool, optional (default=True)) – Whether to print message with early stopping information.
- Returns
callback – The callback that activates early stopping.
- Return type
function