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 a Databricks cluster
  • Train scikit-learn ElasticNet model on a diabetes dataset and log the training metrics, parameters, and model artifacts to a Databricks hosted tracking server
  • View the training results in the MLflow experiment UI

To learn how to deploy the trained model on AWS SageMaker, see scikit-learn model deployment on SageMaker.

MLflow scikit-learn model training notebook