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

Get notebook

scikit-learn model deployment on Azure ML

This notebook uses ElasticNet models trained on the diabetes dataset described in Track scikit-learn model training with MLflow. The notebook shows how to:

• Select a model to deploy using the MLflow experiment UI
• Deploy the model to Azure ML using the MLflow API
• Query the deployed model
• Repeat the deployment and query process for another model
• Delete the deployment using the MLflow API

MLflow scikit-learn model deployment on Azure notebook

Get notebook