fix nthread for XGBoost evaluation

Moelf · Moelf · commit 7e24a6ecf700 · 2023-10-26T07:21:33.000-05:00
diff --git a/src/analysis_utils.jl b/src/analysis_utils.jl
@@ -12,15 +12,15 @@ end
 
 function init_BDT()
     SFinZs = Tuple(
-              XGBoost.Booster(XGBoost.DMatrix[],  model_file = joinpath(BDT_MODEL_DIR, "SFinZ$i.model"))
+              XGBoost.Booster(XGBoost.DMatrix[],  model_file = joinpath(BDT_MODEL_DIR, "SFinZ$i.model"), nthread=1)
               for i = 0:4
              )
     SFnoZs = Tuple(
-              XGBoost.Booster(XGBoost.DMatrix[],  model_file = joinpath(BDT_MODEL_DIR, "SFnoZ$i.model"))
+              XGBoost.Booster(XGBoost.DMatrix[],  model_file = joinpath(BDT_MODEL_DIR, "SFnoZ$i.model"), nthread=1)
               for i = 0:4
              )
     DFs = Tuple(
-              XGBoost.Booster(XGBoost.DMatrix[],  model_file = joinpath(BDT_MODEL_DIR, "DF$i.model"))
+              XGBoost.Booster(XGBoost.DMatrix[],  model_file = joinpath(BDT_MODEL_DIR, "DF$i.model"), nthread=1)
               for i = 0:4
              )
     return function f(ary; fold, region)

Original file line number	Diff line number	Diff line change
`@@ -12,15 +12,15 @@ end`
`12`	`12`
`13`	`13`	`function init_BDT()`
`14`	`14`	`SFinZs = Tuple(`
`15`		`- XGBoost.Booster(XGBoost.DMatrix[], model_file = joinpath(BDT_MODEL_DIR, "SFinZ$i.model"))`
	`15`	`+ XGBoost.Booster(XGBoost.DMatrix[], model_file = joinpath(BDT_MODEL_DIR, "SFinZ$i.model"), nthread=1)`
`16`	`16`	`for i = 0:4`
`17`	`17`	`)`
`18`	`18`	`SFnoZs = Tuple(`
`19`		`- XGBoost.Booster(XGBoost.DMatrix[], model_file = joinpath(BDT_MODEL_DIR, "SFnoZ$i.model"))`
	`19`	`+ XGBoost.Booster(XGBoost.DMatrix[], model_file = joinpath(BDT_MODEL_DIR, "SFnoZ$i.model"), nthread=1)`
`20`	`20`	`for i = 0:4`
`21`	`21`	`)`
`22`	`22`	`DFs = Tuple(`
`23`		`- XGBoost.Booster(XGBoost.DMatrix[], model_file = joinpath(BDT_MODEL_DIR, "DF$i.model"))`
	`23`	`+ XGBoost.Booster(XGBoost.DMatrix[], model_file = joinpath(BDT_MODEL_DIR, "DF$i.model"), nthread=1)`
`24`	`24`	`for i = 0:4`
`25`	`25`	`)`
`26`	`26`	`return function f(ary; fold, region)`