Skip to content

Commit a9a8461

Browse files
authored
[R] use only 1 thread in dmatrix creation (dmlc#11259)
1 parent c95f5d9 commit a9a8461

40 files changed

+118
-117
lines changed

R-package/R/utils.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -485,7 +485,7 @@ xgb.createFolds <- function(y, k) {
485485
#' data(agaricus.train, package = "xgboost")
486486
#'
487487
#' bst <- xgb.train(
488-
#' data = xgb.DMatrix(agaricus.train$data, label = agaricus.train$label),
488+
#' data = xgb.DMatrix(agaricus.train$data, label = agaricus.train$label, nthread = 1),
489489
#' nrounds = 2,
490490
#' params = xgb.params(
491491
#' max_depth = 2,

R-package/R/xgb.Booster.R

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ xgb.get.handle <- function(object) {
265265
#' test <- agaricus.test
266266
#'
267267
#' bst <- xgb.train(
268-
#' data = xgb.DMatrix(train$data, label = train$label),
268+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
269269
#' nrounds = 5,
270270
#' params = xgb.params(
271271
#' max_depth = 2,
@@ -309,7 +309,7 @@ xgb.get.handle <- function(object) {
309309
#' set.seed(11)
310310
#'
311311
#' bst <- xgb.train(
312-
#' data = xgb.DMatrix(as.matrix(iris[, -5]), label = lb),
312+
#' data = xgb.DMatrix(as.matrix(iris[, -5], nthread = 1), label = lb),
313313
#' nrounds = 10,
314314
#' params = xgb.params(
315315
#' max_depth = 4,
@@ -332,7 +332,7 @@ xgb.get.handle <- function(object) {
332332
#' set.seed(11)
333333
#'
334334
#' bst <- xgb.train(
335-
#' data = xgb.DMatrix(as.matrix(iris[, -5]), label = lb),
335+
#' data = xgb.DMatrix(as.matrix(iris[, -5], nthread = 1), label = lb),
336336
#' nrounds = 10,
337337
#' params = xgb.params(
338338
#' max_depth = 4,
@@ -664,7 +664,7 @@ validate.features <- function(bst, newdata) {
664664
#' train <- agaricus.train
665665
#'
666666
#' bst <- xgb.train(
667-
#' data = xgb.DMatrix(train$data, label = train$label),
667+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
668668
#' nrounds = 2,
669669
#' params = xgb.params(
670670
#' max_depth = 2,
@@ -771,7 +771,7 @@ xgb.attributes <- function(object) {
771771
#' train <- agaricus.train
772772
#'
773773
#' bst <- xgb.train(
774-
#' data = xgb.DMatrix(train$data, label = train$label),
774+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
775775
#' nrounds = 2,
776776
#' params = xgb.params(
777777
#' max_depth = 2,
@@ -825,7 +825,7 @@ xgb.config <- function(object) {
825825
#' train <- agaricus.train
826826
#'
827827
#' bst <- xgb.train(
828-
#' data = xgb.DMatrix(train$data, label = train$label),
828+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
829829
#' nrounds = 2,
830830
#' params = xgb.params(
831831
#' max_depth = 2,
@@ -1276,7 +1276,7 @@ xgb.is.same.Booster <- function(obj1, obj2) {
12761276
#' train <- agaricus.train
12771277
#'
12781278
#' bst <- xgb.train(
1279-
#' data = xgb.DMatrix(train$data, label = train$label),
1279+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
12801280
#' nrounds = 2,
12811281
#' params = xgb.params(
12821282
#' max_depth = 2,

R-package/R/xgb.DMatrix.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@
9999
#' )
100100
#' fname <- file.path(tempdir(), "xgb.DMatrix.data")
101101
#' xgb.DMatrix.save(dtrain, fname)
102-
#' dtrain <- xgb.DMatrix(fname)
102+
#' dtrain <- xgb.DMatrix(fname, nthread = 1)
103103
#' @export
104104
#' @rdname xgb.DMatrix
105105
xgb.DMatrix <- function(

R-package/R/xgb.DMatrix.save.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
#' dtrain <- with(agaricus.train, xgb.DMatrix(data, label = label, nthread = 2))
1313
#' fname <- file.path(tempdir(), "xgb.DMatrix.data")
1414
#' xgb.DMatrix.save(dtrain, fname)
15-
#' dtrain <- xgb.DMatrix(fname)
15+
#' dtrain <- xgb.DMatrix(fname, nthread = 1)
1616
#' @export
1717
xgb.DMatrix.save <- function(dmatrix, fname) {
1818
if (typeof(fname) != "character")

R-package/R/xgb.create.features.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,10 +67,10 @@
6767
#'
6868
#' # learning with new features
6969
#' new.dtrain <- xgb.DMatrix(
70-
#' data = new.features.train, label = agaricus.train$label
70+
#' data = new.features.train, label = agaricus.train$label, nthread = 1
7171
#' )
7272
#' new.dtest <- xgb.DMatrix(
73-
#' data = new.features.test, label = agaricus.test$label
73+
#' data = new.features.test, label = agaricus.test$label, nthread = 1
7474
#' )
7575
#' bst <- xgb.train(params = param, data = new.dtrain, nrounds = nrounds)
7676
#'

R-package/R/xgb.cv.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@ xgb.cv <- function(params = xgb.params(), data, nrounds, nfold,
310310
#'
311311
#' train <- agaricus.train
312312
#' cv <- xgb.cv(
313-
#' data = xgb.DMatrix(train$data, label = train$label),
313+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
314314
#' nfold = 5,
315315
#' nrounds = 2,
316316
#' params = xgb.params(

R-package/R/xgb.dump.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
#' test <- agaricus.test
3131
#'
3232
#' bst <- xgb.train(
33-
#' data = xgb.DMatrix(train$data, label = train$label),
33+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
3434
#' nrounds = 2,
3535
#' params = xgb.params(
3636
#' max_depth = 2,

R-package/R/xgb.load.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
#' test <- agaricus.test
3232
#'
3333
#' bst <- xgb.train(
34-
#' data = xgb.DMatrix(train$data, label = train$label),
34+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
3535
#' nrounds = 2,
3636
#' params = xgb.params(
3737
#' max_depth = 2,

R-package/R/xgb.model.dt.tree.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
#' data.table::setDTthreads(nthread)
4444
#'
4545
#' bst <- xgb.train(
46-
#' data = xgb.DMatrix(agaricus.train$data, label = agaricus.train$label),
46+
#' data = xgb.DMatrix(agaricus.train$data, label = agaricus.train$label, nthread = 1),
4747
#' nrounds = 2,
4848
#' params = xgb.params(
4949
#' max_depth = 2,

R-package/R/xgb.save.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
#' test <- agaricus.test
4444
#'
4545
#' bst <- xgb.train(
46-
#' data = xgb.DMatrix(train$data, label = train$label),
46+
#' data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
4747
#' nrounds = 2,
4848
#' params = xgb.params(
4949
#' max_depth = 2,

0 commit comments

Comments
 (0)