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

MLflow inference notebook

Open notebook in new tab Copy link for import