From 4a429a7c4f587a68b100318ee7f6b7059ef323ef Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Sun, 5 Aug 2018 07:42:59 -0700 Subject: [PATCH] Add reg:tweedie to supported objectives in XGBoost4J-Spark (#3552) --- .../dmlc/xgboost4j/scala/spark/params/LearningTaskParams.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 bf64cffa8..e5f6b6cc2 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 @@ -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")