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

[jvm-packages] Add rank:ndcg and rank:map to Spark supported objectives (#3697)

Browse Source
This commit is contained in:
Jerry Lin 2018-09-13 09:51:24 -07:00 committed by Nan Zhu
parent 42b108136f
commit 9acd549dc7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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")