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.