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

[jvm-packages] Expose setMissing method in XGBoostClassificationModel / XGBoostRegressionModel (#4643)

Browse Source
This commit is contained in:
Philip Hyunsu Cho 2019-07-07 16:02:44 -07:00 committed by Nan Zhu
parent 1aaf4a679d
commit d333918f5e
2 changed files with 4 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/XGBoostClassifier.scala
View File

@ -247,6 +247,8 @@ class XGBoostClassificationModel private[ml](
def setTreeLimit(value: Int): this.type = set(treeLimit, value) def setTreeLimit(value: Int): this.type = set(treeLimit, value)
def setMissing(value: Float): this.type = set(missing, value)
def setInferBatchSize(value: Int): this.type = set(inferBatchSize, value) def setInferBatchSize(value: Int): this.type = set(inferBatchSize, value)
/** /**

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

@ -242,6 +242,8 @@ class XGBoostRegressionModel private[ml] (
def setTreeLimit(value: Int): this.type = set(treeLimit, value) def setTreeLimit(value: Int): this.type = set(treeLimit, value)
def setMissing(value: Float): this.type = set(missing, value)
def setInferBatchSize(value: Int): this.type = set(inferBatchSize, value) def setInferBatchSize(value: Int): this.type = set(inferBatchSize, value)
/** /**