[jvm-packages] Add rank:ndcg and rank:map to Spark supported objectiv…

…es (dmlc#3697)
sagunb · Sep 13, 2018 · 9acd549 · 9acd549
1 parent 42b1081
commit 9acd549
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...oost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala b/...oost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala
@@ -82,7 +82,7 @@ private[spark] trait LearningTaskParams extends Params {
 private[spark] object LearningTaskParams {
   val supportedObjective = HashSet("reg:linear", "reg:logistic", "binary:logistic",
     "binary:logitraw", "count:poisson", "multi:softmax", "multi:softprob", "rank:pairwise",
-    "reg:gamma", "reg:tweedie")
+    "rank:ndcg", "rank:map", "reg:gamma", "reg:tweedie")
 
   val supportedEvalMetrics = HashSet("rmse", "mae", "logloss", "error", "merror", "mlogloss",
     "auc", "aucpr", "ndcg", "map", "gamma-deviance")