TimeCatcher: Official implementation of "TimeCatcher: A Variational Framework for Volatility-Aware Forecasting of Non-Stationary Time Series". This efficient MLP-based framework integrates a VAE encoder and a volatility-aware module to significantly improve long-term forecasting, especially in high-volatility scenarios.
Follow these steps to reproduce our experiments:
-
Create a new Conda environment with Python 3.10:
conda create -n TimeCatcher python=3.10 -
Install dependencies
pip install -r requirements.txt -
Prepare datasets
- Download pre-processed datasets from:
- Create a dataset directory:
mkdir ./dataset - Place all downloaded files in
./dataset
- Run experiments
Execute the provided training script:bash ./scripts/TimeCatcher_ecl.shAdditional task scripts available in./scripts/
Note: All experiments were conducted on a single NVIDIA RTX 4090 GPU