fix the merge
This commit is contained in:
@@ -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")
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user