Question about the adjacency matrices

YoZhibo / MSGNet

MSGNet: Learning Multi-Scale Inter-Series Correlations for Multivariate Time Series Forecasting (AAAI2024)

102 stars 17 forks source link

Question about the adjacency matrices #5

Closed Kivirrr closed 4 months ago

Kivirrr commented 4 months ago

Hi, many thanks for contributing such a impressive work! I'm curious about the way you obtained the adjacency matrices you showed for the flight dataset. May I apply these on another multivariable forecasting dataset?

YoZhibo commented 4 months ago

Hi, thanks for your attention. You can simply retrieve the values of the trained adjacency matrix and utilize plt to visualize it. For ease of retrieval, you may set the parameter is_training to 0. Additionally, this approach is generally applicable to other multivariate datasets as well.