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

@@ -64,7 +64,6 @@ object XGBoost extends Serializable {
tracker.getWorkerEnvs.asScala, numWorkers, round, obj, eval)
// force the job
boosters.foreachPartition(_ => ())
println("=====finished training=====")
val booster = boosters.first()
val returnVal = tracker.waitFor()
logger.info(s"Rabit returns with exit code $returnVal")

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)