[jvm-packages] Add back reg:linear for scala. (#4490)

* Add back reg:linear for scala. * Fix linter.
dmlc · May 23, 2019 · 0ce300e · 0ce300e
1 parent 278562d
commit 0ce300e
Showing 1 changed file with 3 additions and 3 deletions.
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
@@ -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")