search

trainingl / STG4Traffic

Code of “STG4Traffic: A Survey and Benchmark of Spatial-Temporal Graph Neural Networks for Traffic Prediction”.
54 stars 10 forks source link

Where are the files? #6

Open dixinyinli2 opened 2 months ago

dixinyinli2 commented 2 months ago

Excuse me, I found this in your code, but I can not find the files. I want to know where are the files. Thanks a lot! dataset_dir = "../data/PEMS-BAY/processed/" graph_pkl = "../data/PEMS-BAY/processed/adj_mx.pkl"

WaldenBlues commented 2 months ago

The project has links about PEMS-Bay dataset but not involved in the repo files. As for missing files or wrong dir problem, (If dataset exists) some may needed to be processed by generate_data.py (the reason for PEMS-Bay/processed/ dir existed)

trainingl commented 2 months ago

Excuse me, I found this in your code, but I can not find the files. I want to know where are the files. Thanks a lot! dataset_dir = "../data/PEMS-BAY/processed/" graph_pkl = "../data/PEMS-BAY/processed/adj_mx.pkl"

Yeah, you need to get the raw data source like METR. Then preprocess the data by using TrafficSpeed/lib/generate_data.py and TrafficSpeed/lib/generate_adj_mx.py files.

trainingl commented 2 months ago

Excuse me, I found this in your code, but I can not find the files. I want to know where are the files. Thanks a lot! dataset_dir = "../data/PEMS-BAY/processed/" graph_pkl = "../data/PEMS-BAY/processed/adj_mx.pkl"

If you still have questions, it would be more convenient to contact me by email.

dixinyinli2 commented 1 month ago

Excuse me, I found this in your code, but I can not find the files. I want to know where are the files. Thanks a lot! dataset_dir = "../data/PEMS-BAY/processed/" graph_pkl = "../data/PEMS-BAY/processed/adj_mx.pkl"

Yeah, you need to get the raw data source like METR. Then preprocess the data by using TrafficSpeed/lib/generate_data.py and TrafficSpeed/lib/generate_adj_mx.py files.

Thank you, I will try it!