search

XDZhelheim / STAEformer

[CIKM'23] Official code for our paper "Spatio-Temporal Adaptive Embedding Makes Vanilla Transformer SOTA for Traffic Forecasting".
https://arxiv.org/abs/2308.10425
132 stars 16 forks source link

Index.npz请教 #11

Closed Andy-HuY closed 3 months ago

Andy-HuY commented 3 months ago

作者您好，请问如果我想试试其他的数据集，这里如果需要读取Index.npz我该如何生成这样的文件呢？谢谢

XDZhelheim commented 3 months ago

你好！ 可以参考：

  1. https://github.com/XDZhelheim/Torch-MTS/blob/main/scripts/generate_training_data.py
  2. https://github.com/zezhishao/BasicTS/blob/master/scripts/data_preparation/METR-LA/generate_training_data.py
Andy-HuY commented 3 months ago

好的，谢谢

---原始邮件--- 发件人: "Zheng @.> 发送时间: 2024年7月30日(周二) 上午10:27 收件人: @.>; 抄送: @.**@.>; 主题: Re: [XDZhelheim/STAEformer] Index.npz请教 (Issue #11)

你好！ 可以参考：

https://github.com/XDZhelheim/Torch-MTS/blob/main/scripts/generate_training_data.py

https://github.com/zezhishao/BasicTS/blob/master/scripts/data_preparation/METR-LA/generate_training_data.py

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>