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

[jvm-packages] removed old group_data from spark api (#3451)

Browse Source
This commit is contained in:
ngoyal2707 2018-07-07 22:21:01 -07:00 committed by Nan Zhu
parent aa90e5c6ce
commit cb017d0c9a
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala
View File

@ -55,13 +55,6 @@ private[spark] trait LearningTaskParams extends Params {
final def getEvalMetric: String = $(evalMetric) 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. * Fraction of training points to use for testing.
*/ */
@ -82,7 +75,7 @@ private[spark] trait LearningTaskParams extends Params {
final def getNumEarlyStoppingRounds: Int = $(numEarlyStoppingRounds) 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) trainTestRatio -> 1.0, numEarlyStoppingRounds -> 0)
} }