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

Backport doc fixes that are compatible with 0.72 release

Browse Source 
* Added python doc string for nthreads to dmatrix (#3363)
* Add instruction to compile XGBoost4J multi-threaded on OSX (#3228)
* Params confusion fixed: num_round, scalePosWeight (#3386)
* Added detailed instruction for adding XGBoost4J as Maven dependency (#3374)
This commit is contained in:
ngoyal2707
2018-06-07 19:16:30 -07:00
committed by Philip Cho
parent 23bc3fc4aa
commit d0f45bede0
5 changed files with 74 additions and 9 deletions
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
* 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" +
" 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
.filter(!_.contains("label")))
.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")
)).setFeaturesCol("features").setLabelCol("label")
// separate