End-to-end example using scikit-learn on Databricks
Note
The managed MLflow integration with Databricks on Google Cloud requires Databricks Runtime for Machine Learning 9.1 LTS or above.
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.