Model inference example

Model inference

This notebook uses an ElasticNet model trained on the diabetes dataset described in Train a scikit-learn model and save in scikit-learn format. 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