beginner-sketch / GMRL

Learning Gaussian Mixture Representations for Tensor Time Series Forecasting

This is a implementation of our GMRL model, which has been accepted to IJCAI 2023.

Installation Dependencies

Python 3 (>= 3.6; Anaconda Distribution)

PyTorch (>= 1.6.0)

Numpy >= 1.17.4

Pandas >= 1.0.3

torch-summary (>= 1.4.5)

Model Training

python main.py -mode train -version 0 cuda_name

Model Evaluation

python main.py -mode eval cuda_name