Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle OMP_THREAD_LIMIT. (#7390)

Browse Source
This commit is contained in:
Jiaming Yuan
2021-11-03 15:44:38 +08:00
committed by GitHub
parent e6ab594e14
commit 57a4b4ff64
4 changed files with 93 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
R-package/R/xgboost.R
View File

@@ -9,8 +9,8 @@ xgboost <- function(data = NULL, label = NULL, missing = NA, weight = NULL,
early_stopping_rounds = NULL, maximize = NULL,
save_period = NULL, save_name = "xgboost.model",
xgb_model = NULL, callbacks = list(), ...) {
dtrain <- xgb.get.DMatrix(data, label, missing, weight, nthread = params$nthread)
merged <- check.booster.params(params, ...)
dtrain <- xgb.get.DMatrix(data, label, missing, weight, nthread = merged$nthread)
watchlist <- list(train = dtrain)