TTS-Norm: Forecasting Tensor Time Series via Multi-way Normalization

This is a implementation of TTS-Norm.

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 cuda_name

Model Evaluation

python main.py -mode eval cuda_name