MLflow

MLflow로 Haystack 파이프라인을 트레이싱하는 방법을 알아봐요. MLflow는 Haystack에 기본 트레이싱을 지원해서, 한 줄의 코드로 모든 파이프라인과 컴포넌트에서 트레이스를 수집할 수 있어요.

출처: 문서

본문

MLflow는 종단 간 머신러닝·AI 수명주기를 관리하는 오픈소스 플랫폼이에요. MLflow는 Haystack에 기본 트레이싱 지원을 제공해서, 한 줄의 코드로 모든 파이프라인과 컴포넌트의 트레이스를 수집할 수 있어요.

MLflow를 설치하세요:

pip install mlflow

모든 Haystack 파이프라인과 컴포넌트의 자동 트레이싱을 활성화해요:

import mlflow

mlflow.haystack.autolog()
# Optionally set an experiment name
mlflow.set_experiment("Haystack")

이렇게 하면 지연 시간, 토큰 사용량, 비용, 그리고 모든 예외를 포함해 모든 Haystack 파이프라인과 컴포넌트의 트레이스를 자동으로 수집해요.

참고: 예시가 포함된 전체 워크스루는 MLflow Haystack 통합 가이드를 확인하세요.