ml-pipeline-setup

Orchestrate end-to-end ML pipelines on Databricks with MLflow and Unity Catalog.

5|6|Updated Jan 26, 2026
One-click install
npx skills add https://github.com/databricks-solutions/vibe-coding-workshop-template --skill ml-pipeline-setup
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: ml-pipeline-setup
Source: https://github.com/databricks-solutions/vibe-coding-workshop-template/tree/main/data_product_accelerator/skills/ml/00-ml-pipeline-setup
Command: npx skills add https://github.com/databricks-solutions/vibe-coding-workshop-template --skill ml-pipeline-setup

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires databricks-asset-bundles, databricks-python-imports, naming-tagging-standards, databricks-autonomous-operations, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill streamlines the creation of robust, production-grade Machine Learning pipelines on Databricks, ensuring consistency and reliability from feature engineering to model deployment.

Core Features & Use Cases

  • End-to-End ML Workflow: Covers feature table creation, model training, and batch inference.
  • Feature Engineering Integration: Leverages Databricks Feature Store for training-serving consistency.
  • MLflow & Unity Catalog: Integrates seamlessly with MLflow for tracking and Unity Catalog for model registry.
  • Use Case: Automate the entire process of building and deploying a predictive model, from raw data in the Gold layer to generating predictions on new data, ensuring best practices are followed at each step.

Quick Start

Use the ml-pipeline-setup skill to create feature tables for your cost data.

Frequently Asked Questions about ml-pipeline-setup

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I build a production machine learning pipeline on Databricks?

To build an end-to-end machine learning pipeline on Databricks, you orchestrate feature engineering, model training, and batch inference while managing MLflow experiment tracking and Unity Catalog model registration. This ensures reproducibility and scalability across the entire workflow.

How does MLflow integrate with Unity Catalog for model registry?

MLflow integrates with Unity Catalog for model registry by tracking experiments and registering trained models directly within the catalog. This combination enforces production best practices for governance, reproducibility, and scalability throughout the machine learning pipeline.

What's the best way to ensure training-serving consistency for batch inference?

To ensure training-serving consistency for batch inference, leverage the Databricks Feature Store during feature engineering. This integration guarantees that the features used during model training match the features generated during batch inference on new data.

Do I need Databricks Asset Bundles to automate ML workflows?

Yes, you need Databricks Asset Bundles to automate ML workflows because they serve as a core dependency for structuring and deploying the pipeline components. Bundles work alongside naming-tagging standards and autonomous operations to maintain production-grade consistency.

Can I use this approach to generate predictions from Gold layer data?

Yes, you can use this approach to generate predictions from Gold layer data. The workflow automates the entire process of building and deploying a predictive model, moving from raw data in the Gold layer to generating predictions on new data while following best practices.

When do I need Databricks Feature Store for feature engineering?

You need the Databricks Feature Store for feature engineering when creating feature tables that require strict training-serving consistency. It integrates directly into the machine learning pipeline to ensure that feature definitions remain consistent from model training through batch inference.