explicitly set nr_cpus=1 when using xgboost

statistikat · Dec 4, 2023 · e8fa390 · e8fa390
1 parent 06c213d
commit e8fa390
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/R/simCategorical.R b/R/simCategorical.R
@@ -555,7 +555,14 @@ simCategorical <- function(simPopObj, additional,
       }else{
         xgb_verbose <- 0
       }
-
+
+      # set nr_cpus=1 for xgboost
+      if(verbose){
+        cat("\n Setting nr_cpus=1 when using xgboost\n")
+      }
+      nr_cpus <- 1
+
+
       if(!dataS@ispopulation){
         weight_str <- paste0("as.numeric(dataSample$", dataS@weight, ")")
         # xgb_weight <- paste0(", info = list(\"weight\" = (",weight_str," - min(", weight_str, "))

diff --git a/R/simContinuous.R b/R/simContinuous.R
@@ -1265,6 +1265,12 @@ simContinuous <- function(simPopObj, additional = "netIncome",
         xgb_verbose <- 0
       }
 
+      # set nr_cpus=1 for xgboost
+      if(verbose){
+        cat("\n Setting nr_cpus=1 when using xgboost\n")
+      }
+      nr_cpus <- 1
+
       if(TRUE){
         xgb_weight <- paste0(", info = list(\"weight\" = as.numeric(dataSample$", weight, "))")
       }else{

diff --git a/src/calibPop.cpp b/src/calibPop.cpp
@@ -195,7 +195,7 @@ IntegerVector calibPop_work(IntegerMatrix inp, NumericVector totals, IntegerVect
     cooldown = cooldown + 1;
     if ( (obj_new <= eps) | (cooldown == 500) ) {
       if ( verb ) {
-        Rprintf("Required precision reached!\nValue of objective function: %g (required precision=%g)\nNumber of iterations=%g", obj_new, eps,counter_all);
+        Rprintf("Required precision reached!\nValue of objective function: %g (required precision=%g)\nNumber of iterations=%f", obj_new, eps,counter_all);
       }
       break;
     }