def _transform(dgl_graph: dgl.DGLGraph, label: torch.Tensor):
dgl_graph.ndata['feat'] = dgl_graph.ndata['attr']
dgl_graph.ndata['node_label'] = dgl_graph.ndata['label']
del dgl_graph.ndata['attr']
del dgl_graph.ndata['label']
static_graph = _conversion.dgl_graph_to_general_static_graph(
dgl_graph)
static_graph.data['label'] = label
return static_graph
def __init__(self, path: str):
dgl_dataset = dgl.data.RedditDataset(
raw_dir=os.path.join(path, '_dgl'))
dgl_graph: dgl.DGLGraph = dgl_dataset[0]
super(RedditDataset, self).__init__(
[_conversion.dgl_graph_to_general_static_graph(dgl_graph)])
def __init__(self, path: str):
dgl_dataset = dgl.data.PubmedGraphDataset(os.path.join(path, '_dgl'))
dgl_graph: dgl.DGLGraph = dgl_dataset[0]
super(PubMedDataset, self).__init__(
[_conversion.dgl_graph_to_general_static_graph(dgl_graph)])
