search

shenyangHuang / TGB

Temporal Graph Benchmark project repo
https://tgb.complexdatalab.com/
182 stars 36 forks source link

tgbl-flight_val_ns_v2.pkl / tgbl-flight_test_ns_v2.pkl creation source code Not Found. #77

Closed HorHang closed 9 months ago

HorHang commented 9 months ago

I would like to create subsample tgbl-flight graph to train the model on my small machine before scaling up to whole graph. However, whole graph tgbl-flight_val_ns_v2.pkl / tgbl-flight_test_ns_v2.pkl is coming with the datasets without code. can you please share the source code?

HorHang commented 9 months ago

I found it by running python tgb\datasets\dataset_scripts\tgbl-flight_neg_generator.py