赞
踩
本文主要关注PyG官方提供的数据集中,因为数据下载源是Dropbox,所以不能直接下载的解决方案。
由于CSDN无法直接显示百度网盘链接,因此放到了GitHub上:PolarisRisingWar/HGNN_Collection: A Collection for HGNN (Heterogenous Graph Neural Network), including datasets, algorithms and so on.,本篇提及的数据集,目录都在dataset - PyG_Dropbox下。
最近更新时间:2022.9.22
最早更新时间:2022.9.22
from torch_geometric.datasets import AMiner
dataset = AMiner(root='/data/pyg_data/AMiner')
data = dataset[0]
这个数据集需要下载的两个文件分别是label.zip和net_aminer.zip,我都已放到百度网盘。
下载后将两个压缩文件分别解压,net_aminer.zip解压后得到net_aminer里面的文件都放到root文件夹下新建的raw文件夹,label.zip解压后的文件都放到root文件夹下新建的raw文件夹。
也就是说,最后raw文件夹下需要有这些文件/文件夹:
'id_author.txt', 'id_conf.txt', 'paper.txt', 'paper_author.txt','paper_conf.txt', 'label'
from torch_geometric.datasets import DBLP
dataset = DBLP(root='/data/pyg_data/DBLP')
data = dataset[0]
这个数据集需要下载的文件是DBLP_processed.zip,我已放到百度网盘。
将文件解压后的文件都放到root文件夹下的raw文件夹即可。
from torch_geometric.datasets import IMDB
dataset = IMDB(root='/data/pyg_data/IMDB')
data = dataset[0]
这个数据集需要下载的文件是IMDB_processed.zip,我已放到百度网盘。
将文件解压后的文件都放到root文件夹下的raw文件夹即可。
from torch_geometric.datasets import LastFM
dataset = LastFM(root='/data/pyg_data/LastFM')
data = dataset[0]
这个数据集需要下载的文件是LastFM_processed.zip,我已放到百度网盘。
将文件解压后的文件都放到root文件夹下的raw文件夹即可。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。