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)])