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

Add reg:tweedie to supported objectives in XGBoost4J-Spark (#3552)

Browse Source
This commit is contained in:
Philip Hyunsu Cho 2018-08-05 07:42:59 -07:00 committed by Nan Zhu
parent 7fefd6865d
commit 4a429a7c4f
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:gamma", "reg:tweedie")
val supportedEvalMetrics = HashSet("rmse", "mae", "logloss", "error", "merror", "mlogloss",
"auc", "aucpr", "ndcg", "map", "gamma-deviance")