Skip to main content

10. Predictive Analytics

You'll understand how Databricks connects feature engineering, model training, deployment, and inference into a governed production lifecycle.

Prereqs: 7. Build the first pipeline, Unity Catalog foundations, 14. CI/CD and DevOps

Why this matters

A Databricks MLOps solution is three Ops disciplines working in concert:

LayerWhat it governsDatabricks tool
DataOpsFeature tables, training sets, inference outputsUnity Catalog
ModelOpsExperiment tracking, model registry, model servingMLflow
DevOpsRepo structure, CI/CD pipelines, environment promotionDABs + Git (see 14. CI/CD and DevOps)

Starter Journey Progress

CI/CD and DevOps
Data Access Control
Automation & Orchestration
Unified AnalyticsDA
Predictive AnalyticsML
AgentsAI
Query and Explore
Build the First Pipeline
Access Your Data
Data Governance Strategy
Cost Monitoring
Infra SetupDEVSTGPRD
Completed Current Pending

Mental model

The diagram below shows the three Ops layers in action. DataOps (Unity Catalog) provides the governed data assets in Prod and Dev catalogs. ModelOps (MLflow) tracks experiments and moves models from Challenger to Champion. DevOps wires it all together through Git. At the end of the day, everything here is code sitting in a repo.

MLOps architecture on Databricks. Unity Catalog governs Dev and Prod catalogs, MLflow handles model training and validation, and Git stores all project code.

Watch first

This video is the definitive guide for MLOps on Databricks. It puts into practice the concepts from the Data Governance, Build the first pipeline, and CI/CD sections. Watch it before you read the subpages.

Deploy the MLOps quickstart

The mlops-quickstart repo trains, registers, and promotes a model end to end. Before you deploy it, get familiar with 14. CI/CD and DevOps. The quickstart uses DABs and GitHub Actions, so understanding those first makes the deployment straightforward.

Once you have the project running, read the Adapt this template with Genie Code section in the repo to swap in your own data, model, and inference pipelines using the built-in Databricks Assistant skills.

Essential reads

ResourceWhat it covers
Manage model lifecycle in Unity CatalogEnd-to-end reference for the full MLOps lifecycle on Databricks
Get started with MLflow 3 for modelsSetup guide and example notebook for MLflow 3
MLflow 3 ML workflowClassic ML training and tracking with MLflow 3
MLflow 3 DL workflowDeep learning training and tracking with MLflow 3

In this section

  • Save a Model to Unity Catalog: train and register a model in UC, and register existing models from a Volume or Hugging Face.
  • Batch Inference: score data with the challenger model in pandas and Spark batch jobs, and run large-scale tracked inference.
  • Datasets as Feature Tables: install the Feature Store and Online Inference demo to explore feature engineering, online tables, and Lakeflow Declarative Pipelines.

Next