echo-ji/st-ssl

ST-SSL (STSSL): Spatio-Temporal Self-Supervised Learning for Traffic Flow Forecasting/Prediction

st-ssl's Issues

can't load other datasets in ST-SSL_Dataset

I download all datasets from ST-SSL_Dataset.
when I load BJTaxi in ST-SSL_Dataset：
Failed to interpret file '../ST-SSL_Dataset/NYCBike1/test.npz' as a pickle.

my numpy version is 1.23.3

[Data Preprocessing] Questions about the data preprocessing procedure.

I would greatly appreciate it if you could elaborate on how to process the dataset.

In the Datasets section, it says that all datasets are processed as a sliding window view, and the format is composed of 4 numpy.ndarray objects.

Could you explain what these "x,y,x_offset,y_offset" mean? or better yet, release the preprocessing code.

Thank you very much for your time and attention to my inquiries.

How to change dataset?

Hi,I want to ask that how to run the projects with another dataset. Now,I can successfully run the projects with the dataset NYCBike 1,but i don't know how to change it into BJTaxi.

ValueError: Cannot load file containing pickled data when allow_pickle=False

Hi，I have a question to ask.
When I run with taxibj dataset, have a bug 'ValueError: Cannot load file containing pickled data when allow_pickle=False' in the line 'cat_data = np.load(os.path.join(data_dir, dataset, category + '.npz'))',
and then I modify this line to 'cat_data = np.load(os.path.join(data_dir, dataset, category + '.npz'), allow_pickle=True)', but still have a bug 'pickle.UnpicklingError: Failed to interpret file 'data/BJTaxi/train.npz' as a pickle';
and also I replace np.load to np.loadtxt, but still unsuccessful.

Could you tell me how to load the taxibj dataset?
thank you!!!

数据集的一点问题

请问作者raw dataset有吗，我想对数据集重新进行处理来适配我的模型，但我在网上并没有找到数据集的原始版本，在您提供的数据集中划分的是19个时间片，我想拿到原始数据来重新划分一个sample含有更多的时间片，因为我看到x_offset中的数据是[[-73][-72][-71][-70][-69][-49][-48][-47][-46][-45][-25][-24][-23][-22][-21][ -3][ -2][ -1][ 0]]，我感觉直接用第一维度乘以第二维度来获得连续的总的时间片是不太靠谱的，所以想请教下您应该怎么做呢？谢谢

RuntimeError

File "D:\tfp_pro\ST-SSL-main\model\aug.py", line 54, in aug_topology drop_prob = torch.softmax(sim_mx[edge_mask], dim=0) RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
clone到本地运行遇到了这个问题

File "D:\AICode\ST-SSL\model\aug.py", line 53, in aug_topology drop_prob = torch.softmax(sim_mx[edge_mask], dim=0) RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

I encountered the following problem when running the program, I debugging found that the sim_mx is in cpu, edge_mask is in gpu, and then I change the edge_mask to CPU, and the program can run successfully, i want to ensure whether changing this place is right?

my change is : edge_mask = (input_graph > 0).tril(diagonal=-1).cpu()

Where can I download the four datasets?

Hello, where can I download the four datasets including NYCBike1, I want to download the datasets from the relevant links of STResNet and STDN, but the relevant links are not working.