Add case for LongParam. (#4885)

To support specifying long parameter as String, the same as other basic type, such as Int, Double ...
2019-09-25 20:41:53 +08:00 · 2019-09-25 20:41:53 +08:00 · 82ee2317e8
commit 82ee2317e8
parent b8433c455a
1 changed files with 2 additions and 0 deletions
--- 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
@ -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)
      }