pySpark:无法保存ML模型

时间:2019-12-06 07:13:10

标签: python-3.x pyspark linear-regression

我无法使用pySpark保存ML模型。

我在打电话

lrModel.save(sc, "lrModel")

我的sc和lrModel被声明为

from pyspark import SparkContext

sc =SparkContext.getOrCreate()
lrModel = lr.fit(train)

我收到此错误。

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-2-a0f9077564b4> in <module>
----> 1 lrModel.save(sc, "lrModel")

TypeError: save() takes 2 positional arguments but 3 were given

我也尝试了不带sc参数的情况。

lrModel.save("lrModel")

它给我以下错误,

\appdata\local\programs\python\python37\lib\site-packages\pyspark\ml\util.py in save(self, path)
    181         if not isinstance(path, basestring):
    182             raise TypeError("path should be a basestring, got type %s" % type(path))
--> 183         self._jwrite.save(path)
    184 
    185     def overwrite(self):

知道我在做什么错吗?我正在使用线性回归模型。

谢谢

0 个答案:

没有答案