From 9acd549dc7e9f21c94046019c889578f60e337ee Mon Sep 17 00:00:00 2001 From: Jerry Lin Date: Thu, 13 Sep 2018 09:51:24 -0700 Subject: [PATCH] [jvm-packages] Add rank:ndcg and rank:map to Spark supported objectives (#3697) --- .../dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 e5f6b6cc2..6f7a32653 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 @@ -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")