search

yutong-xia / CaST

PyTorch implementation of CaST, NeurIPS-23.
22 stars 0 forks source link

How to get dataset? #2

Closed RuiAN77 closed 4 months ago

RuiAN77 commented 5 months ago

Hello, Thank you for this project and contribution. When I try to run the code, I encounter problems with missing datasets such as "adj_mx_pems08.pkl", "time_dalay_attr.pkl", "sparse_adj.npy", "peacor_adj.npy" and "dist_adj.npy". how do I get these dataset? Thanks!

Dx3281101 commented 4 months ago

Hello, Thank you for this project and contribution. When I try to run the code, I encounter problems with missing datasets such as "adj_mx_pems08.pkl", "time_dalay_attr.pkl", "sparse_adj.npy", "peacor_adj.npy" and "dist_adj.npy". how do I get these dataset? Thanks!

Did you solve this problem? I would be grateful if you could share the data file.

yutong-xia commented 4 months ago

Thanks for your interest in our work. These are used to generate edge attributes. You can design your own attribute or generate these files based on the code I provide at https://github.com/yutong-xia/CaST/issues/1#issuecomment-2016405187. For more details about how we built the edge attributes refer to Appendix D of our paper. Thanks!