From e2f09db77a4ea1f759f9be585ab01a320d933900 Mon Sep 17 00:00:00 2001 From: Nan Zhu Date: Wed, 25 Jul 2018 19:57:40 -0700 Subject: [PATCH] [jvm-packages] minor fix for parameter name in example (#3507) --- .../dmlc/xgboost4j/scala/example/spark/SparkWithDataFrame.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jvm-packages/xgboost4j-example/src/main/scala/ml/dmlc/xgboost4j/scala/example/spark/SparkWithDataFrame.scala b/jvm-packages/xgboost4j-example/src/main/scala/ml/dmlc/xgboost4j/scala/example/spark/SparkWithDataFrame.scala index d02ba2fbd..788850459 100644 --- a/jvm-packages/xgboost4j-example/src/main/scala/ml/dmlc/xgboost4j/scala/example/spark/SparkWithDataFrame.scala +++ b/jvm-packages/xgboost4j-example/src/main/scala/ml/dmlc/xgboost4j/scala/example/spark/SparkWithDataFrame.scala @@ -47,7 +47,7 @@ object SparkWithDataFrame { "max_depth" -> 2, "objective" -> "binary:logistic", "num_round" -> numRound, - "nWorkers" -> args(1).toInt).toMap + "num_workers" -> args(1).toInt).toMap val xgboostModel = new XGBoostClassifier(paramMap).fit(trainDF) // xgboost-spark appends the column containing prediction results xgboostModel.transform(testDF).show()