change the API name since we support not only HDFS and local file system

This commit is contained in:
CodingCat
2016-03-11 10:00:37 -05:00
parent 8e3ce908fe
commit 43d7a85bc9
4 changed files with 6 additions and 5 deletions

View File

@@ -40,6 +40,6 @@ object DistTrainWithSpark {
"objective" -> "binary:logistic").toMap
val xgboostModel = XGBoost.train(trainRDD, paramMap, numRound)
// save model to HDFS path
xgboostModel.saveModelToHadoop(outputModelPath)
xgboostModel.saveModelAsHadoopFile(outputModelPath)
}
}