dmlc--dgl
Minjie Wang
f13b9b6212
* WIP: api * dgl.sampling, dgl.data * dgl.sampling; dgl.dataloading * sampling packages * convert * subgraph * deprecate * subgraph APIs * All docstrings for convert/subgraph/transform * almost all funcs under dgl namespace * WIP: DGLGraph * done graph query * message passing functions * lint * fix merge error * fix test * lint * fix Co-authored-by: Quan Gan <coin2028@hotmail.com>
51 行
1.9 KiB
Python
51 行
1.9 KiB
Python
"""The ``dgl.data`` package contains datasets hosted by DGL and also utilities
|
|
for downloading, processing, saving and loading data from external resources.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from . import citation_graph as citegrh
|
|
from .citation_graph import CoraBinary, CitationGraphDataset
|
|
from .minigc import *
|
|
from .tree import SST, SSTDataset
|
|
from .utils import *
|
|
from .sbm import SBMMixture, SBMMixtureDataset
|
|
from .reddit import RedditDataset
|
|
from .ppi import PPIDataset, LegacyPPIDataset
|
|
from .tu import TUDataset, LegacyTUDataset
|
|
from .gnn_benckmark import AmazonCoBuy, CoraFull, Coauthor, AmazonCoBuyComputerDataset, \
|
|
AmazonCoBuyPhotoDataset, CoauthorPhysicsDataset, CoauthorCSDataset, CoraFullDataset
|
|
from .karate import KarateClub, KarateClubDataset
|
|
from .gindt import GINDataset
|
|
from .bitcoinotc import BitcoinOTC, BitcoinOTCDataset
|
|
from .gdelt import GDELT, GDELTDataset
|
|
from .icews18 import ICEWS18, ICEWS18Dataset
|
|
from .qm7b import QM7b, QM7bDataset
|
|
from .dgl_dataset import DGLDataset, DGLBuiltinDataset
|
|
from .citation_graph import CoraGraphDataset, CiteseerGraphDataset, PubmedGraphDataset
|
|
from .knowledge_graph import FB15k237Dataset, FB15kDataset, WN18Dataset
|
|
from .rdf import AIFBDataset, MUTAGDataset, BGSDataset, AMDataset
|
|
|
|
|
|
def register_data_args(parser):
|
|
parser.add_argument(
|
|
"--dataset",
|
|
type=str,
|
|
required=False,
|
|
help=
|
|
"The input dataset. Can be cora, citeseer, pubmed, syn(synthetic dataset) or reddit"
|
|
)
|
|
|
|
|
|
def load_data(args):
|
|
if args.dataset == 'cora':
|
|
return citegrh.load_cora()
|
|
elif args.dataset == 'citeseer':
|
|
return citegrh.load_citeseer()
|
|
elif args.dataset == 'pubmed':
|
|
return citegrh.load_pubmed()
|
|
elif args.dataset is not None and args.dataset.startswith('reddit'):
|
|
return RedditDataset(self_loop=('self-loop' in args.dataset))
|
|
else:
|
|
raise ValueError('Unknown dataset: {}'.format(args.dataset))
|