From b0dcce48065968aa05655e12a1fe39c50f7a540d Mon Sep 17 00:00:00 2001 From: James Lamb Date: Tue, 24 Sep 2019 22:59:52 -0500 Subject: [PATCH] added new early_stopping alias --- R-package/R/lgb.cv.R | 2 +- R-package/R/lgb.train.R | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/R-package/R/lgb.cv.R b/R-package/R/lgb.cv.R index 0699eb20cc26..835a363c5858 100644 --- a/R-package/R/lgb.cv.R +++ b/R-package/R/lgb.cv.R @@ -213,7 +213,7 @@ lgb.cv <- function(params = list(), # If early stopping was passed as a parameter in params(), prefer that to keyword argument # early_stopping_rounds by overwriting the value in 'early_stopping_rounds' - early_stop <- c("early_stopping_round", "early_stopping_rounds", "early_stopping") + early_stop <- c("early_stopping_round", "early_stopping_rounds", "early_stopping", "n_iter_no_change") early_stop_param_indx <- names(params) %in% early_stop if (any(early_stop_param_indx)) { first_early_stop_param <- which(early_stop_param_indx)[[1]] diff --git a/R-package/R/lgb.train.R b/R-package/R/lgb.train.R index e8599d637bfa..aab6fd5229f9 100644 --- a/R-package/R/lgb.train.R +++ b/R-package/R/lgb.train.R @@ -196,7 +196,7 @@ lgb.train <- function(params = list(), # If early stopping was passed as a parameter in params(), prefer that to keyword argument # early_stopping_rounds by overwriting the value in 'early_stopping_rounds' - early_stop <- c("early_stopping_round", "early_stopping_rounds", "early_stopping") + early_stop <- c("early_stopping_round", "early_stopping_rounds", "early_stopping", "n_iter_no_change") early_stop_param_indx <- names(params) %in% early_stop if (any(early_stop_param_indx)) { first_early_stop_param <- which(early_stop_param_indx)[[1]] @@ -205,8 +205,8 @@ lgb.train <- function(params = list(), } using_early_stopping <- !is.null(early_stopping_rounds) - if (using_early_stopping && identical(params$boosting, "dart")){ - warning("Early stopping is not available in 'dart' mode") + if (identical(params$boosting, "dart")){ + warning("Early stopping is not available in 'dart' mode.") use_early_stopping <- FALSE }