"docs/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "2e209c30cf6f2ba42001d0629dc6b7ce354b9a9d"
Unverified Commit f1e7060b authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[GraphBolt] add ogb-lsc-mag240m into BuiltinDataset (#6368)

parent ed3840fc
...@@ -508,6 +508,14 @@ class BuiltinDataset(OnDiskDataset): ...@@ -508,6 +508,14 @@ class BuiltinDataset(OnDiskDataset):
.. note:: .. note::
Reverse edges are added to the original graph. Reverse edges are added to the original graph.
**ogb-lsc-mag240m**
The ogb-lsc-mag240m dataset is a heterogeneous academic graph extracted
from the Microsoft Academic Graph (MAG). See more details in
`ogb-lsc-mag240m <https://ogb.stanford.edu/docs/lsc/mag240m/>`_.
.. note::
Reverse edges are added to the original graph.
Parameters Parameters
---------- ----------
name : str name : str
...@@ -517,7 +525,12 @@ class BuiltinDataset(OnDiskDataset): ...@@ -517,7 +525,12 @@ class BuiltinDataset(OnDiskDataset):
""" """
_base_url = "https://data.dgl.ai/dataset/graphbolt/" _base_url = "https://data.dgl.ai/dataset/graphbolt/"
_datasets = ["ogbn-mag", "ogbl-citation2", "ogbn-products"] _datasets = [
"ogbn-mag",
"ogbl-citation2",
"ogbn-products",
"ogb-lsc-mag240m",
]
def __init__(self, name: str, root: str = "datasets") -> OnDiskDataset: def __init__(self, name: str, root: str = "datasets") -> OnDiskDataset:
dataset_dir = os.path.join(root, name) dataset_dir = os.path.join(root, name)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment