PyG (PyTorch Geometric) Dropbox系图数据集无法下载的解决方案(AMiner, DBLP, IMDB, LastFM)(持续更新ing...)

诸神缄默不语-个人CSDN博文目录

本文主要关注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

文章目录

  • 1. AMiner
  • 2. DBLP
  • 3. IMDB
  • 4. LastFM

1. AMiner

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'

2. DBLP

from torch_geometric.datasets import DBLP

dataset = DBLP(root='/data/pyg_data/DBLP')
data = dataset[0]

这个数据集需要下载的文件是DBLP_processed.zip,我已放到百度网盘。
将文件解压后的文件都放到root文件夹下的raw文件夹即可。

3. IMDB

from torch_geometric.datasets import IMDB

dataset = IMDB(root='/data/pyg_data/IMDB')
data = dataset[0]

这个数据集需要下载的文件是IMDB_processed.zip,我已放到百度网盘。
将文件解压后的文件都放到root文件夹下的raw文件夹即可。

4. LastFM

from torch_geometric.datasets import LastFM

dataset = LastFM(root='/data/pyg_data/LastFM')
data = dataset[0]

这个数据集需要下载的文件是LastFM_processed.zip,我已放到百度网盘。
将文件解压后的文件都放到root文件夹下的raw文件夹即可。

你可能感兴趣的:(人工智能学习笔记,GNN,PyG,Dropbox,AMiner,图神经网络)