MCP HubMCP Hub
스킬 목록으로 돌아가기

orchestrate-ml-pipeline

pjt222
업데이트됨 6 days ago
19 조회
17
2
17
GitHub에서 보기
기타aiautomationdata

정보

이 스킬은 Prefect나 Airflow를 사용하여 DAG 구성, 작업 의존성, 스케줄링, 재시도 로직을 처리하며, 엔드투엔드 머신러닝 파이프라인을 조정합니다. MLflow, DVC, 기능 저장소와 통합되어 프로덕션 워크플로우를 지원합니다. 데이터 수집부터 배포까지의 다단계 ML 프로세스를 자동화하거나 주기적인 모델 재학습을 스케줄링할 때 사용하세요.

빠른 설치

Claude Code

추천
기본
npx skills add pjt222/agent-almanac -a claude-code
플러그인 명령대체
/plugin add https://github.com/pjt222/agent-almanac
Git 클론대체
git clone https://github.com/pjt222/agent-almanac.git ~/.claude/skills/orchestrate-ml-pipeline

Claude Code에서 이 명령을 복사하여 붙여넣어 스킬을 설치하세요

문서

編 ML 線

詳例見 Extended Examples

立全 ML 線—依管、排、察。

  • 自由攝至部之多步 ML 流
  • 排期重訓
  • 協分布處+訓
  • 立 ML 線階間之繁依
  • 管重試+敗復
  • 察線行+敗警
  • 編特工、訓、評、部
  • 立可重之 ML 流跨境

  • :ML 線件(攝、預處、訓、評)
  • :編框擇(Prefect、Airflow、Kubeflow)
  • :Python+編庫
  • :Kubernetes 群為分布
  • :MLflow 跡伺
  • :DVC 為料版
  • :Slack/郵警
  • :察基(Prometheus、Grafana)

一:擇裝編框

擇宜框、立基。

# Option 1: Prefect (modern, Pythonic, simpler)
pip install prefect
pip install prefect-aws prefect-dask prefect-docker

# Start Prefect server (local development)
prefect server start

# Or use Prefect Cloud (managed)
# ... (see EXAMPLES.md for complete implementation)

Airflow 之 Docker Compose:

# docker-compose.airflow.yml
version: '3.8'

x-airflow-common: &airflow-common
  image: apache/airflow:2.8.0
  environment:
    AIRFLOW__CORE__EXECUTOR: CeleryExecutor
    AIRFLOW__DATABASE__SQL_ALCHEMY_CONN: postgresql+psycopg2://airflow:airflow@postgres/airflow
# ... (see EXAMPLES.md for complete implementation)

得: 框裝、UI 達(Prefect http://localhost:4200、Airflow http://localhost:8080)、庫初、排器行。

敗: 查埠(netstat -tulpn | grep 8080)、驗庫連、確 Redis 行(Celery 須)、Python 版(Airflow 須 ≥ 3.8)、Docker daemon、看初日誌。

二:以 Prefect 立 ML 線

各階為 task 之 flow:

# prefect_ml_pipeline.py
from prefect import flow, task
from prefect.tasks import task_input_hash
from datetime import timedelta
import pandas as pd
import mlflow
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
# ... (see EXAMPLES.md for complete implementation)

部+排:

# deploy_prefect.py
from prefect.deployments import Deployment
from prefect.server.schemas.schedules import CronSchedule
from prefect_ml_pipeline import ml_training_pipeline

# Create deployment with schedule
deployment = Deployment.build_from_flow(
    flow=ml_training_pipeline,
# ... (see EXAMPLES.md for complete implementation)

得: Prefect flow 序行諸 task、敗自重試、成則 UI 綠、MLflow 錄驗、模註+部。

敗: 查依正確、MLflow 達、料路正、無循依、task 限期、Prefect 日誌、資源(記憶/CPU)。

三:以 Airflow 立 ML 線

立生產 DAG:

# dags/ml_training_dag.py
from airflow import DAG
from airflow.operators.python import PythonOperator
from airflow.providers.docker.operators.docker import DockerOperator
from airflow.utils.dates import days_ago
from datetime import datetime, timedelta
import mlflow
import pandas as pd
# ... (see EXAMPLES.md for complete implementation)

得: DAG 現於 UI、排行於時、敗觸試+警、XCom 傳料、MLflow 錄驗。

敗: 查 DAG 文法(python dags/ml_training_dag.py)、Airflow 環有引、XCom 不超限(大料用文路)、郵配置、排器行、UI 看 task 日誌。

四:施進階

加動 DAG、分支、並行:

# advanced_pipeline.py (Prefect)
from prefect import flow, task
from prefect.task_runners import DaskTaskRunner, ConcurrentTaskRunner
import time

@task
def process_shard(shard_id: int, data: list) -> dict:
    """Process data shard in parallel."""
# ... (see EXAMPLES.md for complete implementation)

Airflow 分支:

# Airflow branching with BranchPythonOperator
from airflow.operators.python import BranchPythonOperator

def check_data_quality(**context):
    """Decide which branch to take."""
    data_path = context['ti'].xcom_pull(key='data_path')
    df = pd.read_csv(data_path)

# ... (see EXAMPLES.md for complete implementation)

得: 並行 task 同行(線速)、條件分支按理執、動 task 生效、Dask 群分。

敗: Dask 群已配且達、task_runner 已指、分支返有效 task ID、並行無資源爭、條件邏輯正。

五:整察+警

加全察+敗報:

# monitoring_integration.py
from prefect.blocks.notifications import SlackWebhook
from prefect import flow, task, get_run_logger
from prefect.context import FlowRunContext

@task(on_failure=[notify_failure])
def critical_task():
    """Task with failure notification."""
# ... (see EXAMPLES.md for complete implementation)

Airflow 之 sensor 察:

# Airflow SLA and monitoring
from airflow.sensors.base import BaseSensorOperator
from airflow.utils.decorators import apply_defaults

default_args = {
    'sla': timedelta(hours=4),  # Alert if task exceeds 4 hours
    'on_failure_callback': slack_alert_failure,
    'on_success_callback': slack_alert_success,
# ... (see EXAMPLES.md for complete implementation)

得: 敗送 Slack/郵、SLA 違觸警、自指追、日誌聚於察系。

敗: 驗 Slack webhook 正、郵 SMTP 設、警塊已載、SLA 值合理、無網阻警。

六:施線之 CI/CD

控版+自部署:

# .github/workflows/deploy-pipeline.yml
name: Deploy ML Pipeline

on:
  push:
    branches: [main]
    paths:
      - 'pipelines/**'
# ... (see EXAMPLES.md for complete implementation)

得: 部前試過、自部於生產、部成告隊、Git 控線版。

敗: 試覆+敗、Prefect Cloud 憑證、部腳本處誤、Slack webhook、CI 日誌。

  • 編框裝且行
  • 線 DAG 立、依正
  • 諸 task 序行
  • 重試於敗起作用
  • 排運於時
  • MLflow 整錄驗
  • DVC 整版料
  • 並行 task 同行
  • 條件分支正
  • 察+警有效
  • CI/CD 自部
  • 線跨境可重

  • 循依:A 依 B、B 依 A—慎設 DAG、用 Airflow/Prefect 驗器
  • 記憶漏:長 task 累記憶—設限期、察用、期重啟工
  • XCom 限:傳大料於 XCom—用文路或外存(S3)
  • 時區惑:排運於誤時—皆用 UTC、明設時區
  • 缺重試:暫誤致 task 永敗—配指數退之重試
  • 緊耦:task 直依實作—用清介、明傳參
  • 無冪等:重行致重複或誤—設冪等(重試安)
  • 誤理弱:敗無境—加詳日誌、捕例
  • 資源爭:並行壓資源—限並、設配額
  • 版衝突:諸 task 須不容依—用 Docker 隔

  • track-ml-experiments
  • version-ml-data
  • build-feature-store
  • deploy-ml-model-serving
  • deploy-to-kubernetes

GitHub 저장소

pjt222/agent-almanac
경로: i18n/wenyan-ultra/skills/orchestrate-ml-pipeline
0
agentsagentskillsai-assisted-developmentclaude-codeskillsteams

연관 스킬

llamaguard

기타

LlamaGuard는 폭력 및 혐오 발언 등 6가지 안전 범주에서 LLM 입력과 출력을 조정하기 위한 Meta의 70-80억 파라미터 모델입니다. 94-95% 정확도를 제공하며 vLLM, Hugging Face 또는 Amazon SageMaker를 사용해 배포할 수 있습니다. 이 기술을 사용하여 AI 애플리케이션에 콘텐츠 필터링 및 안전 가드레일을 손쉽게 통합하세요.

스킬 보기

cost-optimization

기타

이 Claude Skill은 리소스 적정화, 태깅 전략, 지출 분석을 통해 개발자들이 클라우드 비용을 최적화할 수 있도록 지원합니다. AWS, Azure, GCP에서 클라우드 비용을 절감하고 비용 거버넌스를 구현하기 위한 프레임워크를 제공합니다. 인프라 비용을 분석하거나, 리소스를 적정화하거나, 예산 제약을 충족해야 할 때 사용하세요.

스킬 보기

quantizing-models-bitsandbytes

기타

이 스킬은 bitsandbytes를 사용하여 LLM을 8비트 또는 4비트 정밀도로 양자화하며, 최소한의 정확도 손실로 50-75%의 메모리 감소를 달성합니다. 제한된 GPU 메모리에서 더 큰 모델을 실행하거나 추론을 가속화하는 데 이상적이며, INT8, NF4, FP4와 같은 형식을 지원합니다. 이 스킬은 HuggingFace Transformers와 통합되어 QLoRA 학습 및 8비트 옵티마이저를 가능하게 합니다.

스킬 보기

dispatching-parallel-agents

기타

이 Claude Skill은 3개 이상의 독립적인 문제를 동시에 조사하고 해결하기 위해 다중 에이전트를 배치합니다. 공유 상태나 의존성 없이 해결 가능한 무관련 장애 시나리오에 맞게 설계되었습니다. 핵심 기능은 병렬 문제 해결로, 각 독립 문제 영역마다 하나의 에이전트를 할당하여 효율성을 극대화합니다.

스킬 보기