---
title: "Tutorial: End-to-end classic ML models on Databricks | Databricks on AWS"
source: https://docs.databricks.com/aws/en/mlflow/end-to-end-example
ingestedAt: "2026-06-18T08:13:58.705Z"
---

This tutorial notebook presents an end-to-end example of training a classic ML model in Databricks, including loading data, visualizing the data, setting up a parallel hyperparameter optimization, and using MLflow to review the results, register the model, and perform inference on new data using the registered model in a Spark UDF.

You can [import this notebook](https://docs.databricks.com/aws/en/notebooks/notebook-export-import) and run it yourself, or copy code-snippets and ideas for your own use.

## Notebook[​](#notebook "Direct link to Notebook")

*   MLflow 3
*   MLflow 2.x

This version of the notebook uses [MLflow 3](https://docs.databricks.com/aws/en/mlflow/mlflow-3-install) and Unity Catalog.

#### XGBoost MLflow 3 tutorial (Unity Catalog)
