cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results for 
Search instead for 
Did you mean: 

Cannot serialize this model error when attempting MLFlow for SparkNLP

Slalom_Tobias
New Contributor III

I'm attempting to use MLFlow to register models in Databricks and am following the recipe at...

https://nlp.johnsnowlabs.com/docs/en/licensed_serving_spark_nlp_via_api_databricks_mlflow

when i execute...

mlflow.spark.log_model(pipeline, "lemmatizer", conda_env=conda_env)

i get the following error....

MlflowException: Cannot serialize this model. MLflow can only save descendants of pyspark.Modelthat implement MLWritable and MLReadable.

can anyone help?

3 REPLIES 3

Kaniz_Fatma
Community Manager
Community Manager

Hi @Tobias Cortese​, Please go through the documentation. It might give you some insights into how to help resolve your question.

Vidula
Honored Contributor

Hi @Tobias Cortese​ 

Hope all is well! Just wanted to check in if you were able to resolve your issue and would you be happy to share the solution or mark an answer as best? Else please let us know if you need more help. 

We'd love to hear from you.

Thanks!

Slalom_Tobias
New Contributor III

I haven't found a solution within the documentation.