THUDM/HGB

关于NC/benchmark/methods/RSHN/data的数据下载链接失效

Closed this issue · 9 comments

您好,我想要测试RSHN中的三个NC数据集(ACM, DBLP, IMDB),但是Read.me中给的链接已经失效。可以麻烦您提供可下载的链接么

所有链接应该都有清华云盘和google drive两份,清华云盘因为我毕业了所以失效了,只能用google drive了。

google drive也只是提供了三个link预测的数据集,并没有提供NC预测的数据集(ACM, DBLP, IMDB)

NC数据集需要到NC目录里:https://github.com/THUDM/HGB/tree/master/NC

如果你说的是HGB benchmark数据集,就需要到benchmark目录里:https://github.com/THUDM/HGB/tree/master/NC/benchmark/methods/RSHN

您好,我就是在NC目录里找的。但是只有
微信图片_20240125104657
这是RSHN在link预测用到的数据集。我想要的是node classification中的ACM, DBLP, IMDB数据集

那就在主页的readme里:https://github.com/THUDM/HGB

您是说这个链接么Node Classification: https://drive.google.com/drive/folders/10-pf2ADCjq_kpJKFHHLHxr_czNNCJ3aX?usp=sharing
这个链接提供的也是用于HGB这类的节点分类数据集格式。RSHN需要的是这类格式才可以
def load_rel_graph(name):
name = name
x_path = '/HGB/NC/benchmark/methods/RSHN/data/data/' + name + '-relation/rel_graph_x.npy'
edge_index_path = '/HGB/NC/benchmark/methods/RSHN/data/data/' + name + '-relation/rel_graph_edge_index.npy'
edge_attr_path = '/HGB/NC/benchmark/methods/RSHN/data/data/' + name + '-relation/rel_graph_edge_attr.npy'

那就是这个:

如果你说的是HGB benchmark数据集,就需要到benchmark目录里:https://github.com/THUDM/HGB/tree/master/NC/benchmark/methods/RSHN

RSHN原文并没有评测ACM、DBLP、IMDB,要评测这几个数据集就要用我们这个代码。我们修改了RSHN的数据加载代码以支持我们的数据格式。

抱歉,我的疏忽。我再检查代码时发现RSHN的关系数据加载是需要运行代码之后再生成的。问题解决了,感谢您的及时回复