added init_model

mlr3learners · Apr 4, 2020 · 2c6683c · 2c6683c · lintr-bot · Apr 4, 2020
1 parent c588db2
commit 2c6683c
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/R/LearnerClassifLightGBM.R b/R/LearnerClassifLightGBM.R
@@ -31,6 +31,9 @@ LearnerClassifLightGBM = R6::R6Class(
                        default = 5L,
                        lower = 3L,
                        tags = c("config", "train")),
+          ParamUty$new(id = "init_model",
+                       default = NULL,
+                       tags = c("config", "train")),
           #######################################
           #######################################
           # Classification only
@@ -557,6 +560,8 @@ LearnerClassifLightGBM = R6::R6Class(
       self$param_set$values[["nrounds_by_cv"]] = NULL
       nfolds = self$param_set$values[["nfolds"]]
       self$param_set$values[["nfolds"]] = NULL
+      init_model = self$param_set$values[["init_model"]]
+      self$param_set$values[["init_model"]] = NULL
       # get training parameters
       pars = self$param_set$get_values(tags = "train")
       # train CV model, in case that nrounds_by_cv is true
@@ -574,6 +579,7 @@ LearnerClassifLightGBM = R6::R6Class(
           , nfold = nfolds
           , stratified = TRUE
           , eval = feval
+          , init_model  = init_model
         )
         message(
           sprintf(
@@ -593,6 +599,7 @@ LearnerClassifLightGBM = R6::R6Class(
         , data = private$dtrain
         , params = pars
         , eval = feval
+        , init_model  = init_model
       ) # use the mlr3misc::invoke function (it's similar to do.call())
     },
     .predict = function(task) {

diff --git a/R/LearnerRegrLightGBM.R b/R/LearnerRegrLightGBM.R
@@ -29,6 +29,9 @@ LearnerRegrLightGBM = R6::R6Class(
                          default = 5L,
                          lower = 3L,
                          tags = c("config", "train")),
+            ParamUty$new(id = "init_model",
+                         default = NULL,
+                         tags = c("config", "train")),
             #######################################
             #######################################
             # Regression only

diff --git a/R/lgbparams.R b/R/lgbparams.R
@@ -15,6 +15,9 @@ lgbparams = function() {
                    default = 5L,
                    lower = 3L,
                    tags = c("config", "train")),
+      ParamUty$new(id = "init_model",
+                   default = NULL,
+                   tags = c("config", "train")),
       #######################################
       #######################################
       # Regression only

diff --git a/vignettes/mlr3learners_lightgbm_binary.Rmd b/vignettes/mlr3learners_lightgbm_binary.Rmd
@@ -88,7 +88,9 @@ learner$param_set$values = mlr3misc::insert_named(
     "learning_rate" = 0.1,
     "seed" = 17L,
     "metric" = "auc",
-    "num_iterations" = 100
+    "num_iterations" = 100,
+    "snapshot_freq" = 10,
+    "output_model" = "LGB.txt"
   )
 )
 ```