unit8co/darts

A python library for user-friendly forecasting and anomaly detection on time series.

무엇을 해결하는가

Darts는 시계열 예측 및 이상 탐지를 단순화하기 위해 설계된 Python 라이브러리입니다. 전통적인 통계 방법부터 고급 딥러닝 아키텍처에 이르기까지 광범위한 모델에 대해 통합된 인터페이스를 제공하여, 다양한 예측 전략을 구현하고 비교하는 복잡성을 줄여줍니다.

어떻게 작동하는가

이 라이브러리는 scikit-learn과 유사한 일관된 fit()predict() API를 사용하여 사용자가 모델을 쉽게 교체할 수 있도록 합니다. 단변량 및 다변량 시계열을 모두 지원하며, 외부 데이터(과거 및 미래 공변량)와 정적 공변량을 처리할 수 있습니다. 딥러닝 모델은 PyTorch Lightning을 사용하여 구현되어 GPU/TPU 가속 및 사용자 정의 학습 루프를 지원합니다. 이상 탐지의 경우, Darts는 전용 스코어러와 탐지기를 제공하거나, 기존 예측 모델을 래핑하여 예측값과 실제값 사이의 불일치를 식별할 수 있도록 합니다.

누구를 위한 것인가

여러 시계열을 포함하는 잠재적으로 대규모 데이터 세트에서 시계열 분석, 예측 및 이상 탐지를 수행해야 하는 데이터 과학자 및 ML 엔지니어를 대상으로 합니다.

Highlights

  • Diverse Model Suite: ARIMA 및 Exponential Smoothing부터 N-BEATS 및 기타 딥 뉴럴 네트워크를 포함합니다.
  • Multivariate & Global Models: 다변량 시계열을 지원하며, 단일 모델을 여러 다른 시계열에 대해 학습시킬 수 있는 능력을 갖추고 있습니다.
  • Probabilistic Forecasting: 신뢰 구간 및 파라메트릭 분포 또는 분위수 추정을 지원합니다.
  • Comprehensive Tooling: 백테스팅, 데이터 처리(스케일링, 결측치 채우기) 및 계층적 예측 조정(hierarchical forecast reconciliation)을 위한 내장 유틸리티를 포함합니다.
  • Anomaly Detection: 이상치를 점수화하고 탐지하기 위한 통합 도구(PyOD와의 호환성 포함)를 제공합니다.
  • Backend Compatibility: pandas, polars, numpy, pyarrow 및 xarray와 원활하게 통합됩니다.