在 Pyspark 中保存随机森林分类器模型时出错

Question

rf = RandomForestClassifier().setFeaturesCol("features").setLabelCol("label")
pipeline = Pipeline(stages=[tokenizer, hashingTF, idf, rf])
model = pipeline.fit(training) 
model.save(sc, '<path_to_save>')

我正在尝试使用上面的代码保存模型文件。但是出现意外错误 -

TypeError: save() takes exactly 2 arguments (3 given)

我不明白这个错误。我只传递了 2 个参数，但仍然收到此错误。 有人知道吗？我在这里做错了什么？

Answer 1

我不知道这是如何工作的。但是删除第一个字段“sc”对我有用。

model.save('<path_to_save>')

我可以通过这个命令保存模型文件。