DGL数据集MiniGCDataset解读

from dgl.data import MiniGCDataset

数据集包含8种不同类型的图形。

• 第0类：循环图
• 第1类：星形图
• 第2类：车轮图
• 第3类：棒棒糖图
• 第4类：超立方体图
• 第5类：网格图
• 第6类：集团图
• 第7类：圆形梯形图

例如：class 0

注：该数据集类与pytorch的Dataset类兼容。

类: dgl.data.MiniGCDataset（num_graphs，min_num_v，max_num_v )

参数：	num_graphs（int）–此数据集中的图的个数。 min_num_v（int）–所有图中，节点最少的数目 max_num_v（int）–所有图中，节点最多的数目

例如：

#初始化数据集
#dataset数据集一共有80个图，最小的图的节点有10个，最大的图的节点有20个。

dataset = MiniGCDataset(80, 10, 20)

#获取数据集个数,lens=80
lens = MiniGCDataset.__len__(dataset)

’’’取第一个图，graph为一个dgl.DGLGraph类型的图，例如：
DGLGraph(num_nodes=13, num_edges=39,
         ndata_schemes={}
         edata_schemes={})
label为该图的标签
’’’
graph, label = dataset[0]

#第二种方法
mol = MiniGCDataset.__getitem__(dataset,0)
graph, label = mol