search

Starlien95 / GraphPrompt

GraphPrompt: Unifying Pre-Training and Downstream Tasks for Graph Neural Networks
139 stars 14 forks source link

KeyError: 'raw' #10

Closed Lylovewhy closed 10 months ago

Lylovewhy commented 10 months ago

Traceback (most recent call last): File "/home/ylin/GraphPrompt/graphdownstream/pre_train.py", line 462, in data = pretrain_load_data(train_config["graph_dir"], train_config["graphslabel_dir"], num_workers=train_config["num_workers"]) File "/home/ylin/GraphPrompt/graphdownstream/utils.py", line 623, in pretrain_load_data for g, graph in graphs["raw"].items(): KeyError: 'raw'

looking forward to your reply~

Lylovewhy commented 10 months ago
微信图片_20231128193943

../data/ENZYMES下面没有ENZYMESPreTrain，只有../data下有ENZYMESPreTrain

SwaggyZhang commented 10 months ago

微信图片_20231128193943 ../data/ENZYMES下面没有ENZYMESPreTrain，只有../data下有ENZYMESPreTrain

你好，我也在follow这篇工作，此处应该是一个笔误，仅需把这个ENZYMESPreTrain移动到ENZYMES文件夹下就可以了，如果帮助到你，希望能和你一起讨论~

Starlien95 commented 10 months ago

您只需要调整对应文件夹的位置即可，同时我们重新上传了代码