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 8d6bc74ab..076dff42e 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 @@ -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) }