Track scikit-learn model training with MLflow

This notebook is based on the MLflow scikit-learn diabetes tutorial.

The notebook shows how to use MLflow to track the model training process, including logging model parameters, metrics, the model itself, and other artifacts like plots to an Azure Databricks hosted tracking server. It also includes instructions for viewing the logged results in the MLflow tracking UI.

MLflow scikit-learn model training notebook

Get notebook