pySpark:保存ML模型

时间:2016-04-13 23:04:44

标签: apache-spark machine-learning pyspark

有人可以举例说明如何在pySpark中保存ML模型吗?

有关

ml.classification.LogisticRegressionModel

我尝试使用以下内容:

model.save("path")

但它似乎不起作用。

2 个答案:

答案 0 :(得分:1)

如果我正确理解您的问题,则您的方法签名不正确。

根据文档,你还需要传递你的火花背景。

文档:https://spark.apache.org/docs/1.6.1/api/python/pyspark.mllib.html?highlight=save#pyspark.mllib.classification.LogisticRegressionModel.save

答案 1 :(得分:0)

在Spark 2.3.0中,如果您使用的是ML:

model.save("path")

引用:Spark ML model .save

我刚刚运行LogisticRegression并保存了它。


但是,如果您使用的是mllib,则另一个答案建议使用:

save(sc, path)

引用:Spark MLLib model .save