Train a scikit-learn model and save in scikit-learn format

This notebook is based on the MLflow tutorial. The notebook shows how to:

  • Install MLflow on an Azure Databricks cluster
  • Train scikit-learn ElasticNet model on a diabetes dataset and log the training metrics, parameters, and model artifacts to a Azure Databricks hosted tracking server
  • View the training results in the MLflow experiment UI

To learn how to deploy the trained model on Azure ML, see scikit-learn model deployment on Azure ML.

MLflow scikit-learn model training notebook

Get notebook