Add case for LongParam. (#4885)

To support specifying long parameter as String, the same as other basic type, such as Int, Double ...
dmlc · Sep 25, 2019 · 82ee231 · 82ee231
1 parent b8433c4
commit 82ee231
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...s/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala b/...s/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala
@@ -260,6 +260,8 @@ private[spark] trait ParamMapFuncs extends Params {
           set(name, paramValue.toString.toInt)
         case _: FloatParam =>
           set(name, paramValue.toString.toFloat)
+        case _: LongParam =>
+          set(name, paramValue.toString.toLong)
         case _: Param[_] =>
           set(name, paramValue)
       }