adjust the API signature as well as the docs
This commit is contained in:
@@ -56,7 +56,7 @@ object XGBoost {
|
||||
val trainMat = new DMatrix(dataIter, null)
|
||||
val watches = List("train" -> trainMat).toMap
|
||||
val round = 2
|
||||
val booster = XGBoostScala.train(paramMap, trainMat, round, watches, null, null)
|
||||
val booster = XGBoostScala.train(trainMat, paramMap, round, watches, null, null)
|
||||
Rabit.shutdown()
|
||||
collector.collect(new XGBoostModel(booster))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user