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

[jvm-packages] allow training with missing values in xgboost-spark (#1525)

Browse Source 
* allow training with missing values in xgboost-spark

* fix compilation error

* fix bug
This commit is contained in:
Nan Zhu
2016-08-29 21:45:49 -04:00
committed by GitHub
parent 6014839961
commit 3f198b9fef
2 changed files with 34 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -128,7 +128,7 @@ class XGBoostSuite extends FunSuite with BeforeAndAfter {
List("eta" -> "1", "max_depth" -> "6", "silent" -> "0",
"objective" -> "binary:logistic").toMap,
new scala.collection.mutable.HashMap[String, String],
numWorkers = 2, round = 5, null, null, useExternalMemory = false)
numWorkers = 2, round = 5, eval = null, obj = null, useExternalMemory = false)
val boosterCount = boosterRDD.count()
assert(boosterCount === 2)
val boosters = boosterRDD.collect()