From cb017d0c9aa7d8e168d09a79811aea6eb887e99f Mon Sep 17 00:00:00 2001 From: ngoyal2707 Date: Sat, 7 Jul 2018 22:21:01 -0700 Subject: [PATCH] [jvm-packages] removed old group_data from spark api (#3451) --- .../scala/spark/params/LearningTaskParams.scala | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala index e477b242fe8b..bf64cffa843b 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala @@ -55,13 +55,6 @@ private[spark] trait LearningTaskParams extends Params { final def getEvalMetric: String = $(evalMetric) - /** - * group data specify each group sizes for ranking task. To correspond to partition of - * training data, it is nested. - */ - final val groupData = new GroupDataParam(this, "groupData", "group data specify each group " + - "size for ranking task. To correspond to partition of training data, it is nested.") - /** * Fraction of training points to use for testing. */ @@ -82,7 +75,7 @@ private[spark] trait LearningTaskParams extends Params { final def getNumEarlyStoppingRounds: Int = $(numEarlyStoppingRounds) - setDefault(objective -> "reg:linear", baseScore -> 0.5, groupData -> null, + setDefault(objective -> "reg:linear", baseScore -> 0.5, trainTestRatio -> 1.0, numEarlyStoppingRounds -> 0) }