使用MLLib保存线性回归模型时出错

Question

[尝试将线性回归模型保存到磁盘时，我收到此错误：“ TypeError：save（）接受2个位置参数，但给出了3个位置参数”]

从pyspark导入SparkConf，SparkContext

从pyspark.sql导入SQLContext

从pyspark.ml.regression导入LinearRegression

sc = SparkContext（）

lr = LinearRegression（featuresCol ='features'，labelCol ='NextOrderInDays'，maxIter = 10，regParam = 0.3，elasticNetParam = 0.8）

lr_model = lr.fit（train_df）

lr_model.save（sc，“ lr_model.model”）

搜索网络输出的内容类似于我写的内容。作为第三个论点我想念什么？

谢谢

Answer 1

您使用ml软件包而不是mllib：from pyspark.ml.regression import LinearRegression。

因此，保存功能只有一个参数：路径（参见documentation）。