Model inference example
Model inference
This notebook uses an ElasticNet model trained on the diabetes dataset described in Track scikit-learn model training with MLflow. This notebook shows how to:
Select a model to deploy using the MLflow experiment UI
Load the trained model as a scikit-learn model
Create a PySpark UDF from the model
Apply the UDF to add a prediction column to a DataFrame