more updates for Flink

more fix
This commit is contained in:
CodingCat
2016-03-11 10:07:00 -05:00
parent 43d7a85bc9
commit aca0096b33
4 changed files with 6 additions and 5 deletions

View File

@@ -70,7 +70,7 @@ object XGBoost {
* @param modelPath The path that is accessible by hadoop filesystem API.
* @return The loaded model
*/
def loadModelFromHadoop(modelPath: String) : XGBoostModel = {
def loadModelFromHadoopFile(modelPath: String) : XGBoostModel = {
new XGBoostModel(
XGBoostScala.loadModel(
FileSystem

View File

@@ -31,7 +31,7 @@ class XGBoostModel (booster: Booster) extends Serializable {
*
* @param modelPath The model path as in Hadoop path.
*/
def saveModelToHadoop(modelPath: String): Unit = {
def saveModelAsHadoopFile(modelPath: String): Unit = {
booster.saveModel(FileSystem
.get(new Configuration)
.create(new Path(modelPath)))