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

force the user to set number of workers

Browse Source
This commit is contained in:
CodingCat
2016-03-12 13:33:57 -05:00
parent 980898f3fb
commit 16b9e92328
6 changed files with 20 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
doc/jvm/xgboost4j-intro.md
View File

@@ -114,7 +114,7 @@ val trainRDD = MLUtils.loadLibSVMFile(sc, inputTrainPath).repartition(args(1).to
We move forward to train the models:
```scala
val xgboostModel = XGBoost.train(trainRDD, paramMap, numRound)
val xgboostModel = XGBoost.train(trainRDD, paramMap, numRound, numWorkers)
```
@@ -147,7 +147,7 @@ val trainData = MLUtils.readLibSVM(env, "/path/to/data/agaricus.txt.train")
Model Training can be done as follows
```scala
val xgboostModel = XGBoost.train(trainData, paramMap, round)
val xgboostModel = XGBoost.train(trainData, paramMap, round, nWorkers)
```