Ошибка сохранения модели линейной регрессии с MLLib - PullRequest
0 голосов
/ 07 октября 2019

Пытаясь сохранить мою модель линейной регрессии на диск, я получаю эту ошибку: «TypeError: save () принимает 2 позиционных аргумента, но 3 даны» *

из pyspark import SparkConf, SparkContext

из pyspark.sql import SQLContext

из pyspark.ml.regression import LinearRegression

sc = SparkContext ()

lr = LinearRegression (featuresCol = 'features',labelCol = 'NextOrderInDays', maxIter = 10, regParam = 0.3 ,asticNetParam = 0.8)

lr_model = lr.fit (train_df)

lr_model.save (sc, "lr_model.model")

Поиск в сети выдает что-то похожее на то, что я написал. Что мне не хватает в качестве третьего аргумента?

Спасибо

1 Ответ

1 голос
/ 07 октября 2019

Вы используете пакет ml, а не mllib: from pyspark.ml.regression import LinearRegression.

Таким образом, функция сохранения имеет только один аргумент: путь (ср. документация ).

...