Skip to content

Commit 54dcc79

Browse files
committed
fix: default nrounds value
1 parent e471d56 commit 54dcc79

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

R/LearnerClassifXgboost.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,7 @@ LearnerClassifXgboost = R6Class("LearnerClassifXgboost",
425425
#' @export
426426
default_values.LearnerClassifXgboost = function(x, search_space, task, ...) { # nolint
427427
special_defaults = list(
428-
nrounds = 1L
428+
nrounds = 1000L
429429
)
430430
defaults = insert_named(default_values(x$param_set), special_defaults)
431431
defaults[search_space$ids()]

R/LearnerRegrXgboost.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@ LearnerRegrXgboost = R6Class("LearnerRegrXgboost",
310310
#' @export
311311
default_values.LearnerRegrXgboost = function(x, search_space, task, ...) { # nolint
312312
special_defaults = list(
313-
nrounds = 1L
313+
nrounds = 1000L
314314
)
315315
defaults = insert_named(default_values(x$param_set), special_defaults)
316316
defaults[search_space$ids()]

0 commit comments

Comments
 (0)