kevin-xuan / FOGS

[IJCAI'2022] FOGS: First-Order Gradient Supervision with Learning-based Graph for Traffic Flow Forecasting
https://www.ijcai.org/proceedings/2022/545
23 stars 3 forks source link

About learn_graph.py #2

Open melarrr opened 5 months ago

melarrr commented 5 months ago

when i run learn_graph.py to get PEMS04's learngraph i got AttributeError: 'Namespace' object has no attribute 'thresh_T',maybe thresh_cos also need to solve. What should i do ,thank u?

kevin-xuan commented 5 months ago

Hello, sorry for the late reply. Do you follow the steps in the readme file? We first run "graph_preparation.py" to generate time correlation graph, and the parameters in this file include "Thres_T". Then, after learning node embeddings through node2vec algorithm, "learn_graph.py" is used to learn the final graph. The parameter in this file is "Thres_cos". There is no Thres_T parameter.