在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>')

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