Skip to content

GitLab

"git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "616388e01651ff79e7be6aedb98b33e5ca3701a4"
Commit 14a6d94f authored by rgranvil's avatar rgranvil Committed by Guolin Ke
Browse files

fix weight and init_score assignment in lgb.cv (#1040)

parent bc0579c8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
R-package/R/lgb.cv.R
View file @ 14a6d94f
...@@ -236,7 +236,7 @@ lgb.cv <- function(params = list(), ...@@ -236,7 +236,7 @@ lgb.cv <- function(params = list(),
if (!is.list(folds[[1]])) { if (!is.list(folds[[1]])) {
bst_folds <- lapply(seq_along(folds), function(k) { bst_folds <- lapply(seq_along(folds), function(k) {
dtest <- slice(data, folds[[k]]) dtest <- slice(data, folds[[k]])
dtrain <- slice(data, unlist(folds[-k])) dtrain <- slice(data, seq_len(nrow(data))[-folds[[k]]])
setinfo(dtrain, "weight", getinfo(data, "weight")[-folds[[k]]]) setinfo(dtrain, "weight", getinfo(data, "weight")[-folds[[k]]])
setinfo(dtrain, "init_score", getinfo(data, "init_score")[-folds[[k]]]) setinfo(dtrain, "init_score", getinfo(data, "init_score")[-folds[[k]]])
setinfo(dtest, "weight", getinfo(data, "weight")[folds[[k]]]) setinfo(dtest, "weight", getinfo(data, "weight")[folds[[k]]])
......
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