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
- Export the model as a PySpark UDF
- Apply the UDF to add a prediction column to a DataFrame