Skip to content

beginner-sketch/GMRL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages