zezhishao / STEP

Code for our SIGKDD'22 paper Pre-training-Enhanced Spatial-Temporal Graph Neural Network For Multivariate Time Series Forecasting.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hope to add visual part of the code

xiewenqiang666 opened this issue · comments

Can you give the visual code of Figure 3?
BasicTS also does not provide the code of the prediction part, can this part be provided?

commented

Thanks for your attention.
Unfortunately, the scripts of visualization are not tracked by git, and I also can not find these codes now.
You can plot Figure 3(a), (c), and (d) using the similarity metrics here based on seaborn and matplotlib.
And you can plot Figure 1(b) using the returns of TSFormer based on matplotlib.
These are not heavy works.

Thank you very much for your patient guidance.

commented

You're welcome and feel free to create a new issue if you have any questions.