change the API name since we support not only HDFS and local file system
This commit is contained in:
@@ -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)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user