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

[jvm-packages] minor fix for parameter name in example (#3507)

Browse Source
This commit is contained in:
Nan Zhu
2018-07-25 19:57:40 -07:00
committed by GitHub
parent a725272e19
commit e2f09db77a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jvm-packages/xgboost4j-example/src/main/scala/ml/dmlc/xgboost4j/scala/example/spark/SparkWithDataFrame.scala
View File

@@ -47,7 +47,7 @@ object SparkWithDataFrame {
"max_depth" -> 2, "max_depth" -> 2,
"objective" -> "binary:logistic", "objective" -> "binary:logistic",
"num_round" -> numRound, "num_round" -> numRound,
"nWorkers" -> args(1).toInt).toMap "num_workers" -> args(1).toInt).toMap
val xgboostModel = new XGBoostClassifier(paramMap).fit(trainDF) val xgboostModel = new XGBoostClassifier(paramMap).fit(trainDF)
// xgboost-spark appends the column containing prediction results // xgboost-spark appends the column containing prediction results
xgboostModel.transform(testDF).show() xgboostModel.transform(testDF).show()