pytorch-forecasting: 해석 가능한 멀티 호라이즌 시계열 예측을 위한 고수준 딥러닝 프레임워크

해결하는 문제

PyTorch Forecasting은 최첨단 딥러닝 아키텍처를 사용하여 시계열 예측을 위한 고수준 API를 제공합니다. 복잡한 시계열 데이터를 처리하고, 신경망을 훈련하며, 실제 예측 작업을 위해 배포하는 과정을 단순화합니다.

작동 방식

PyTorch Lightning을 기반으로 구축된 이 패키지는 CPU 또는 GPU에서의 훈련 복잡성을 추상화합니다. 변수 변환, 결측치 처리 및 다양한 히스토리 길이를 처리하기 위해 특화된 TimeSeriesDataSet 클래스를 사용합니다. 사용자는 Temporal Fusion Transformers (TFT), N-BEATS, N-HiTS, 그리고 DeepAR과 같은 여러 사전 구현된 신경망 아키텍처 중에서 선택할 수 있으며, Optuna를 통한 하이퍼파라미터 튜닝과 멀티 호라이즌 메트릭을 사용하여 이를 최적화할 수 있습니다.

대상 사용자

시계열 예측에서 최대의 유연성을 추구하는 전문가와 빠르게 시작할 수 있는 합리적인 기본값을 필요로 하는 초보자 모두를 위해 설계되었습니다.

주요 특징

• 최첨단 모델: Temporal Fusion Transformers, N-BEATS, N-HiTS, 그리고 DeepAR의 구현을 포함합니다.
• 데이터 추상화: 시계열 메타데이터와 변환을 관리하기 위한 전용 데이터셋 클래스.
• 확장 가능한 훈련: 자동 로깅 및 하드웨어 간 원활한 확장성을 위해 PyTorch Lightning을 활용합니다.
• 해석 가능성: 내장된 해석 기능이 있는 모델과 실제값 대 예측값에 대한 일반적인 시각화를 포함합니다.