[FLINK] remove nWorker from API

This commit is contained in:
tqchen
2016-03-14 16:18:04 -07:00
parent 084ed6224d
commit 90f7220736
4 changed files with 6 additions and 8 deletions

View File

@@ -33,9 +33,8 @@ object DistTrainWithFlink {
"objective" -> "binary:logistic").toMap
// number of iterations
val round = 2
val nWorkers = 5
// train the model
val model = XGBoost.train(trainData, paramMap, round, 5)
val model = XGBoost.train(trainData, paramMap, round)
val predTest = model.predict(testData.map{x => x.vector})
model.saveModelAsHadoopFile("file:///path/to/xgboost.model")
}