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

params confusion fixed (#3386)

Browse Source
This commit is contained in:
Bruce Qu 2018-06-16 04:17:35 +08:00 committed by Nan Zhu
parent 34e3edfb1a
commit 578a0c7ddb
2 changed files with 2 additions and 2 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/BoosterParams.scala
View File

@ -154,7 +154,7 @@ trait BoosterParams extends Params {
/** /**
* Control the balance of positive and negative weights, useful for unbalanced classes. A typical * Control the balance of positive and negative weights, useful for unbalanced classes. A typical
* value to consider: sum(negative cases) / sum(positive cases). [default=0] * value to consider: sum(negative cases) / sum(positive cases). [default=1]
*/ */
val scalePosWeight = new DoubleParam(this, "scale_pos_weight", "Control the balance of positive" + val scalePosWeight = new DoubleParam(this, "scale_pos_weight", "Control the balance of positive" +
" and negative weights, useful for unbalanced classes. A typical value to consider:" + " and negative weights, useful for unbalanced classes. A typical value to consider:" +

2
jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostSparkPipelinePersistence.scala
View File

@ -90,7 +90,7 @@ class XGBoostSparkPipelinePersistence extends FunSuite with PerTest
.setInputCols(df.columns .setInputCols(df.columns
.filter(!_.contains("label"))) .filter(!_.contains("label")))
.setOutputCol("features") .setOutputCol("features")
val xgbEstimator = new XGBoostEstimator(Map("num_rounds" -> 10, val xgbEstimator = new XGBoostEstimator(Map("num_round" -> 10,
"tracker_conf" -> TrackerConf(60 * 60 * 1000, "scala") "tracker_conf" -> TrackerConf(60 * 60 * 1000, "scala")
)).setFeaturesCol("features").setLabelCol("label") )).setFeaturesCol("features").setLabelCol("label")
// separate // separate