Train and register a Scikit learn model for model serving

This notebook trains an ElasticNet model using the diabetes dataset from scikit learn. Databricks autologging is also used to both log metrics and to register the trained model to the Databricks Model Registry.

After you run this notebook in its entirety, you have a registered model for model serving with Databricks Model Serving (AWS|Azure| GCP).

Import libraries

Load data

Log and register the model

The following code automattically logs the trained model. By specifying registered_model_name in the autologging configuration, the model trained is automatically registered to Unity Catalog.