ml-pipeline

Implements ML pipelines with experiment tracking, orchestration, feature stores, and model validation.

Updated Mar 9, 2026
One-click install
npx skills add https://github.com/ArMaTeC/Redball --skill ml-pipeline-armatec
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: ml-pipeline
Source: https://github.com/ArMaTeC/Redball/tree/main/.devin/skills/ml-pipeline
Command: npx skills add https://github.com/ArMaTeC/Redball --skill ml-pipeline-armatec

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires mlflow, kfp, scikit-learn, pandas, great-expectations, feast, wandb, prefect, apache-airflow, and includes references (resource) components.

What problem does it solve? Building production machine learning systems requires coordinating experiment tracking, training orchestration, feature stores, and model validation, which is error-prone and hard to reproduce when done ad hoc. ## Core Features & Use Cases - Experiment Tracking: Configures MLflow or Weights & Biases logging for parameters, metrics, artifacts, and model registry operations. - Pipeline Orchestration: Creates Kubeflow Pipelines, Airflow DAGs, or Prefect flows with retries, branching, and scheduled retraining. - Feature Engineering & Validation: Builds scikit-learn transformation pipelines, Feast feature stores, and Great Expectations data validation checkpoints. - Use Case: A data science team needs an automated daily retraining workflow that validates incoming data, trains a model, logs metrics to MLflow, and deploys only when accuracy exceeds a threshold. ## Quick Start Use the ml-pipeline skill to build a Kubeflow training pipeline with MLflow experiment tracking and a validation gate before deployment.

Frequently Asked Questions about ml-pipeline

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

FAQPage Schema
How do I set up MLflow experiment tracking for model training?

MLflow tracking logs parameters, metrics, and model artifacts within a run context using mlflow.start_run(). The skill provides wrapper classes for logging hyperparameters, registering models to the MLflow Model Registry, and transitioning versions between Staging and Production stages.

How to build a Kubeflow pipeline for ML training?

Kubeflow pipelines are defined with the kfp.dsl component decorator, where each step (load, preprocess, train, evaluate) is a containerized component passing Dataset and Model artifacts. The skill includes a complete template with conditional deployment using dsl.If based on evaluation thresholds.

Kubeflow vs Airflow vs Prefect for ML pipeline orchestration?

Kubeflow runs containerized components on Kubernetes with native artifact passing, Airflow uses DAGs with XCom for task communication and scheduling, and Prefect offers Python-native flows with caching and retries. The skill provides working templates for all three orchestrators.

How do I detect data drift in ML features?

Data drift detection compares current feature distributions against a reference using the Kolmogorov-Smirnov test or Population Stability Index. The skill's FeatureDriftDetector flags drift when p-values fall below significance levels or PSI exceeds 0.2.

What is shadow deployment for ML models?

Shadow deployment runs a candidate model alongside production, logging predictions without affecting responses. The skill's ShadowDeployment class compares prediction agreement and latency between models, letting you validate performance on real traffic before promotion.

When should I not use a full ML pipeline framework?

Full pipeline orchestration is unnecessary for quick one-off experiments, simple scripts without hyperparameters, or interactive prototyping. The references explicitly recommend against these tools for ad-hoc analysis without reproducibility or deployment requirements.