Skip to content

nataliekoh/GNNs_MultivariateTSForecasting

Repository files navigation

Multivariate Time Series Forecasting with Graph Neural Networks

Natalie Koh, Zachary Laswick, Daiwei Shen

Datasets

Architectures Used

  • STEP
  • Graph WaveNet
  • Simple graph convolutional network with LSTM layer implemented in Keras

Scripts

  • For data pre-processing, see PruneDatasets_SingleSubject.ipynb.
  • To run STEP on the datasets, use scripts in STEP/ModifiedSTEPCode.
  • To run Graph WaveNET, cd into the WaveNet directory and run python train.py --gcn_bool.
  • To run the simple GCN implemented in Keras, use KerasGNNwLSTM.ipynb.

About

Evaluating the performance of STEP with WaveNet and Graph WaveNet architectures on multivariate time series forecasting

Resources

Stars

1 star

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors