[jvm-packages] remove APIs with DMatrix from xgboost-spark (#1519)

* test consistency of prediction functions between DMatrix and RDD

* remove APIs with DMatrix from xgboost-spark

* fix compilation error in xgboost4j-example

* fix test cases
This commit is contained in:
Nan Zhu
2016-08-28 21:25:49 -04:00
committed by GitHub
parent 6d65aae091
commit 74db1e8867
3 changed files with 5 additions and 23 deletions

View File

@@ -47,7 +47,7 @@ object DistTrainWithSpark {
"objective" -> "binary:logistic").toMap
val xgboostModel = XGBoost.train(trainRDD, paramMap, numRound, nWorkers = args(1).toInt,
useExternalMemory = true)
xgboostModel.predict(new DMatrix(testSet))
xgboostModel.booster.predict(new DMatrix(testSet))
// save model to HDFS path
xgboostModel.saveModelAsHadoopFile(outputModelPath)
}