From 0ce300e73ad9c586e41762d07f7acdcfa03f323d Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Fri, 24 May 2019 06:02:08 +0800 Subject: [PATCH] [jvm-packages] Add back reg:linear for scala. (#4490) * Add back reg:linear for scala. * Fix linter. --- .../xgboost4j/scala/spark/params/LearningTaskParams.scala | 6 +++--- 1 file changed, 3 insertions(+), 3 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 f343aab69..f7fd38bed 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 @@ -105,9 +105,9 @@ private[spark] trait LearningTaskParams extends Params { } private[spark] object LearningTaskParams { - val supportedObjective = HashSet("reg:squarederror", "reg:logistic", "binary:logistic", - "binary:logitraw", "count:poisson", "multi:softmax", "multi:softprob", "rank:pairwise", - "rank:ndcg", "rank:map", "reg:gamma", "reg:tweedie") + val supportedObjective = HashSet("reg:linear", "reg:squarederror", "reg:logistic", + "binary:logistic", "binary:logitraw", "count:poisson", "multi:softmax", "multi:softprob", + "rank:pairwise", "rank:ndcg", "rank:map", "reg:gamma", "reg:tweedie") val supportedObjectiveType = HashSet("regression", "classification")