added protection against use of 'cb.early.stop'

microsoft · Sep 29, 2019 · 2049df4 · 2049df4
1 parent 76d00a8
commit 2049df4
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 8 deletions.
diff --git a/R-package/R/lgb.cv.R b/R-package/R/lgb.cv.R
@@ -221,13 +221,25 @@ lgb.cv <- function(params = list(),
     early_stopping_rounds <- params[[first_early_stop_param_name]]
   }
 
-  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")
-    use_early_stopping <- FALSE
+  # Did user pass parameters that indicate they want to use early stopping?
+  using_early_stopping_via_args <- !is.null(early_stopping_rounds)
+
+  # Cannot use early stopping with 'dart' boosting
+  if (identical(params$boosting, "dart")){
+    warning("Early stopping is not available in 'dart' mode.")
+    using_early_stopping_via_args <- FALSE
+
+    # Remove the cb.early.stop() function if it was passed in to callbacks
+    callbacks <- Filter(
+      f = function(cb_func){
+        !identical(attr(cb_func, "name"), "cb.early.stop")
+      }
+      , x = callbacks
+    )
   }
 
-  if (using_early_stopping){
+  # If user supplied early_stopping_rounds, add the early stopping callback
+  if (using_early_stopping_via_args){
     callbacks <- add.cb(
       callbacks
       , cb.early.stop(

diff --git a/R-package/R/lgb.train.R b/R-package/R/lgb.train.R
@@ -204,13 +204,25 @@ lgb.train <- function(params = list(),
     early_stopping_rounds <- params[[first_early_stop_param_name]]
   }
 
-  using_early_stopping <- !is.null(early_stopping_rounds)
+  # Did user pass parameters that indicate they want to use early stopping?
+  using_early_stopping_via_args <- !is.null(early_stopping_rounds)
+
+  # Cannot use early stopping with 'dart' boosting
   if (identical(params$boosting, "dart")){
     warning("Early stopping is not available in 'dart' mode.")
-    use_early_stopping <- FALSE
+    using_early_stopping_via_args <- FALSE
+
+    # Remove the cb.early.stop() function if it was passed in to callbacks
+    callbacks <- Filter(
+      f = function(cb_func){
+        !identical(attr(cb_func, "name"), "cb.early.stop")
+      }
+      , x = callbacks
+    )
   }
 
-  if (using_early_stopping){
+  # If user supplied early_stopping_rounds, add the early stopping callback
+  if (using_early_stopping_via_args){
     callbacks <- add.cb(
       callbacks
       , cb.early.stop(