Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add case for LongParam. (#4885)

Browse Source 
To support specifying long parameter as String, the same as other basic
type, such as Int, Double ...
This commit is contained in:
Liangcai Li 2019-09-25 20:41:53 +08:00 committed by Nan Zhu
parent b8433c455a
commit 82ee2317e8
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala
View File

@ -260,6 +260,8 @@ private[spark] trait ParamMapFuncs extends Params {
set(name, paramValue.toString.toInt) set(name, paramValue.toString.toInt)
case _: FloatParam => case _: FloatParam =>
set(name, paramValue.toString.toFloat) set(name, paramValue.toString.toFloat)
case _: LongParam =>
set(name, paramValue.toString.toLong)
case _: Param[_] => case _: Param[_] =>
set(name, paramValue) set(name, paramValue)
} }