Tutorial: End-to-end ML models on Databricks


The managed MLflow integration with Databricks on Google Cloud requires Databricks Runtime for Machine Learning 9.1 LTS or above.

Machine learning in the real world is messy. Data sources contain missing values, include redundant rows, or may not fit in memory. Feature engineering often requires domain expertise and can be tedious. Modeling too often mixes data science and systems engineering, requiring not only knowledge of algorithms but also of machine architecture and distributed systems.

Databricks simplifies this process. The following 10-minute tutorial notebook shows an end-to-end example of training machine learning models on tabular data. You can import this notebook and run it yourself, or copy code-snippets and ideas for your own use.


The following notebook may include functionality that is not available in this release of Databricks on Google Cloud.


This notebook requires Databricks Runtime 8.1 ML or above.


MLflow end-to-end example notebook

Open notebook in new tab