"...git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "0bbb02fd59cea7bb53408510062ab1b8d67193f8"
Unverified Commit 34b74842 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[R-package] fix CVBooster reset_parameter() method (fixes #4900) (#4901)

* [R-package] fix CVBooster reset_parameter() method (fixes #4900)

* make it clear that there should be one booster per fold
parent 0df38106
...@@ -12,8 +12,8 @@ CVBooster <- R6::R6Class( ...@@ -12,8 +12,8 @@ CVBooster <- R6::R6Class(
return(invisible(NULL)) return(invisible(NULL))
}, },
reset_parameter = function(new_params) { reset_parameter = function(new_params) {
for (x in boosters) { for (x in self$boosters) {
x$reset_parameter(params = new_params) x[["booster"]]$reset_parameter(params = new_params)
} }
return(invisible(self)) return(invisible(self))
} }
......
...@@ -380,6 +380,31 @@ test_that("cv works", { ...@@ -380,6 +380,31 @@ test_that("cv works", {
expect_false(is.null(bst$record_evals)) expect_false(is.null(bst$record_evals))
}) })
test_that("CVBooster$reset_parameter() works as expected", {
dtrain <- lgb.Dataset(train$data, label = train$label)
n_folds <- 2L
cv_bst <- lgb.cv(
params = list(
objective = "regression"
, min_data = 1L
, num_leaves = 7L
, verbose = VERBOSITY
)
, data = dtrain
, nrounds = 3L
, nfold = n_folds
)
expect_is(cv_bst, "lgb.CVBooster")
expect_length(cv_bst$boosters, n_folds)
for (bst in cv_bst$boosters) {
expect_equal(bst[["booster"]]$params[["num_leaves"]], 7L)
}
cv_bst$reset_parameter(list(num_leaves = 11L))
for (bst in cv_bst$boosters) {
expect_equal(bst[["booster"]]$params[["num_leaves"]], 11L)
}
})
test_that("lgb.cv() rejects negative or 0 value passed to nrounds", { test_that("lgb.cv() rejects negative or 0 value passed to nrounds", {
dtrain <- lgb.Dataset(train$data, label = train$label) dtrain <- lgb.Dataset(train$data, label = train$label)
params <- list( params <- list(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment