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

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

Browse Source 
* Add back reg:linear for scala.

* Fix linter.
This commit is contained in:
Jiaming Yuan 2019-05-24 06:02:08 +08:00 committed by Nan Zhu
parent 278562db13
commit 0ce300e73a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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