fix the merge

This commit is contained in:
CodingCat
2016-03-06 15:22:05 -05:00
parent 16008ebfb8
commit 6499422e90
5 changed files with 8 additions and 6 deletions

View File

@@ -130,6 +130,7 @@ class XGBoostSuite extends FunSuite with BeforeAndAfterAll {
trainingRDD,
List("eta" -> "1", "max_depth" -> "2", "silent" -> "0",
"objective" -> "binary:logistic").toMap,
new scala.collection.mutable.HashMap[String, String],
numWorker, 2, null, null)
val boosterCount = boosterRDD.count()
assert(boosterCount === numWorker)