End-to-end example using scikit-learn on Databricks

This notebook uses scikit-learn to illustrate a complete end-to-end example of loading data, model training, distributed hyperparameter tuning, and model inference. It also illustrates how to use MLflow and Model Registry for logging and registering your model. You can import this notebook and run it in your Databricks workspace.

Requirements

Databricks Runtime 7.0 ML or above.

Use scikit-learn with MLflow integration notebook

Open notebook in new tab