From 87aca8c244145848f61dac87654b4490bd190a36 Mon Sep 17 00:00:00 2001 From: Sergei Lebedev Date: Mon, 1 Oct 2018 20:22:01 +0200 Subject: [PATCH] [jvm-packages] Fixed the distributed updater check (#3739) The updater used in distributed training is grow_histmaker and not grow_colmaker as the error message stated prior to this commit. --- .../ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala index f0ad69ad5..782569c51 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala @@ -224,10 +224,10 @@ private[spark] trait ParamMapFuncs extends Params { def XGBoostToMLlibParams(xgboostParams: Map[String, Any]): Unit = { for ((paramName, paramValue) <- xgboostParams) { if ((paramName == "booster" && paramValue != "gbtree") || - (paramName == "updater" && paramValue != "grow_colmaker,prune")) { + (paramName == "updater" && paramValue != "grow_histmaker,prune")) { throw new IllegalArgumentException(s"you specified $paramName as $paramValue," + s" XGBoost-Spark only supports gbtree as booster type" + - " and grow_colmaker,prune as the updater type") + " and grow_histmaker,prune as the updater type") } val name = CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.LOWER_CAMEL, paramName) params.find(_.name == name) match {