Source: eclairjs/ml/util/MLWriter.js

  1. /*                                                                         
  2.  * Copyright 2016 IBM Corp.
  3.  *
  4.  * Licensed under the Apache License, Version 2.0 (the "License");
  5.  * you may not use this file except in compliance with the License.
  6.  * You may obtain a copy of the License at
  7.  *
  8.  *      http://www.apache.org/licenses/LICENSE-2.0
  9.  *
  10.  * Unless required by applicable law or agreed to in writing, software
  11.  * distributed under the License is distributed on an "AS IS" BASIS,
  12.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13.  * See the License for the specific language governing permissions and
  14.  * limitations under the License.
  15.  */
  17. (function () {
  19.     var JavaWrapper = require(EclairJS_Globals.NAMESPACE + '/JavaWrapper');
  20.     var Logger = require(EclairJS_Globals.NAMESPACE + '/Logger');
  21.     var Utils = require(EclairJS_Globals.NAMESPACE + '/Utils');
  24.     /**
  25.      * @classdesc
  26.      * Utility classes that can save ML instances.
  27.      * @class
  28.      * @memberof module:eclairjs/ml/util
  29.      */
  31.     var MLWriter = function (jvmObject) {
  32.         this.logger = Logger.getLogger("ml_util_MLWriter_js");
  33.         JavaWrapper.call(this, jvmObject);
  35.     };
  37.     MLWriter.prototype = Object.create(JavaWrapper.prototype);
  39.     MLWriter.prototype.constructor = MLWriter;
  42.     /**
  43.      * Saves the ML instances to the input path.
  44.      * @param {string} path
  45.      */
  46.     MLWriter.prototype.save = function (path) {
  47.             this.getJavaObject().save(path);
  48.     };
  51.     /**
  52.      * Overwrites if the output path already exists.
  53.      * @returns {module:eclairjs/ml/util.MLWriter}
  54.      */
  55.     MLWriter.prototype.overwrite = function () {
  56.            var javaObject =  this.getJavaObject().overwrite();
  57.            return new MLWriter(javaObject);
  58.     };
  61.     /**
  62.      * @param {module:eclairjs/sql.SQLContext} sqlContext
  63.      * @returns {module:eclairjs/ml/util.MLWriter}
  64.      */
  65.     MLWriter.prototype.context = function (sqlContext) {
  66.            var sqlContext_uw = Utils.unwrapObject(sqlContext);
  67.            var javaObject =  this.getJavaObject().context(sqlContext_uw);
  68.            return new MLWriter(javaObject);
  69.     };
  71.     /**
  72.      * @param {module:eclairjs/sql.SparkSession} sparkSession
  73.      * @returns {module:eclairjs/ml/util.MLWriter} 
  74.      */
  75.     MLWriter.prototype.session = function(sparkSession) {
  76.        var sparkSession_uw = Utils.unwrapObject(sparkSession);
  77.        var javaObject =  this.getJavaObject().session(sparkSession_uw);
  78.        return new MLWriter(javaObject);
  79.     };
  80.     
  82.     module.exports = MLWriter;
  83. })();