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

microsoft · Nov 9, 2017 · 14a6d94 · 14a6d94
1 parent bc0579c
commit 14a6d94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R-package/R/lgb.cv.R b/R-package/R/lgb.cv.R
@@ -236,7 +236,7 @@ lgb.cv <- function(params = list(),
   if (!is.list(folds[[1]])) {
     bst_folds <- lapply(seq_along(folds), function(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, "init_score", getinfo(data, "init_score")[-folds[[k]]])
       setinfo(dtest, "weight", getinfo(data, "weight")[folds[[k]]])