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如何生成 #12

Open Riverdells opened 2 months ago

Riverdells commented 2 months ago

作者您好,请问每个数据集的index.npz如何生成的

Riverdells commented 2 months ago

我想使用文章以外的数据集进行训练,非常感谢如此出色的工作!

XDZhelheim commented 2 months ago

你好! 可以参考:

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

Ref: https://github.com/XDZhelheim/STAEformer/issues/11#issuecomment-2257338580

Andy-HuY commented 1 month ago

我想使用文章以外的数据集进行训练,非常感谢如此出色的工作!

你好,请问是否有具体实践成功