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 )
参数: |
---|
例如:
#初始化数据集
#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