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

[jvm-packages] Fix mis configure of nthread (#1709)

Browse Source 
* add back train method but mark as deprecated

* fix scalastyle error

* change class to object in examples

* fix compilation error

* fix mis configuration
This commit is contained in:
Nan Zhu 2016-10-27 12:10:35 -04:00 committed by GitHub
parent f12074d355
commit d321375df5
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala
View File

@ -24,7 +24,10 @@ private[spark] trait GeneralParams extends Params {
val round = new IntParam(this, "num_round", "The number of rounds for boosting",
ParamValidators.gtEq(1))
val nWorkers = new IntParam(this, "nthread", "number of workers used to run xgboost",
val nWorkers = new IntParam(this, "nworkers", "number of workers used to run xgboost",
ParamValidators.gtEq(1))
val numThreadPerTask = new IntParam(this, "nthread", "number of threads used by per worker",
ParamValidators.gtEq(1))
val useExternalMemory = new BooleanParam(this, "use_external_memory", "whether to use external" +
@ -42,6 +45,7 @@ private[spark] trait GeneralParams extends Params {
val missing = new FloatParam(this, "missing", "the value treated as missing")
setDefault(round -> 1, nWorkers -> 1, useExternalMemory -> false, silent -> 0,
setDefault(round -> 1, nWorkers -> 1, numThreadPerTask -> 1,
useExternalMemory -> false, silent -> 0,
customObj -> null, customEval -> null, missing -> Float.NaN)
}