more updates for Flink

more fix
2016-03-11 10:07:00 -05:00
parent 43d7a85bc9
commit aca0096b33
4 changed files with 6 additions and 5 deletions
--- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostModel.scala
+++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostModel.scala
@@ -49,7 +49,7 @@ class XGBoostModel(booster: Booster)(implicit val sc: SparkContext) extends Seri
   *
   * @param modelPath The model path as in Hadoop path.
   */
-  def saveModelToHadoopFile(modelPath: String): Unit = {
+  def saveModelAsHadoopFile(modelPath: String): Unit = {
    val path = new Path(modelPath)
    val outputStream = path.getFileSystem(sc.hadoopConfiguration).create(path)
    booster.saveModel(outputStream)