# Copyright (C) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE in project root for information.
from mmlspark.vw._VowpalWabbitRegressor import _VowpalWabbitRegressor, _VowpalWabbitRegressionModel
from pyspark.ml.common import inherit_doc
[docs]@inherit_doc
class VowpalWabbitRegressor(_VowpalWabbitRegressor):
def _create_model(self, java_model):
model = VowpalWabbitRegressionModel()
model._java_obj = java_model
model._transfer_params_from_java()
return model
[docs]@inherit_doc
class VowpalWabbitRegressionModel(_VowpalWabbitRegressionModel):
[docs] def saveNativeModel(self, filename):
"""
Save the native model to a local or WASB remote location.
"""
self._java_obj.saveNativeModel(filename)