如何将pyspark ml模型保存到泡菜文件中:SPARK MLLIB

时间:2018-11-18 09:50:23

标签: python-3.x machine-learning deep-learning apache-spark-mllib

我在sparkmllib中训练了一个模型。现在,我要将其保存在泡菜文件中。

我使用了下面的过程,但是它以不同的格式存储。

pipeline = Pipeline(stages=[tokenizer,hashTF,hashIDF,nbClass])

prediction = pipeline.fit(training_df)

prediction.save("path")

model = PipelineModel.load("path")

pre = model.transform(testing_df)

pre.show()

0 个答案:

没有答案