Answers for "load saved model pyspark"

1

load saved model pyspark

print(spark.version)
2.4.3

# fit model
cvModel = cv_grid.fit(train_df)

# save best model to specified path
mPath =  "/path/to/model/folder"
cvModel.bestModel.write().overwrite().save(mPath)

# read pickled model via pipeline api
from pyspark.ml.pipeline import PipelineModel
persistedModel = PipelineModel.load(mPath)

# predict
predictionsDF = persistedModel.transform(test_df)
Posted by: Guest on May-29-2021

Code answers related to "load saved model pyspark"

Browse Popular Code Answers by Language