[Graphbolt][Dataset] Add igb-hom dataset (#7781)

Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com>
This commit is contained in:
Bowen Yao
2024-09-06 00:04:53 -05:00
committed by GitHub
parent 12841c675b
commit f9097ee438
5 changed files with 13 additions and 6 deletions

View File

@@ -340,6 +340,7 @@ def parse_args():
"igb-hom-small",
"igb-hom-medium",
"igb-hom-large",
"igb-hom",
],
)
parser.add_argument("--root", type=str, default="datasets")

View File

@@ -371,10 +371,11 @@ def parse_args():
"igb-hom-small",
"igb-hom-medium",
"igb-hom-large",
"igb-hom",
],
help="The dataset we can use for node classification example. Currently"
" ogbn-products, ogbn-arxiv, ogbn-papers100M and"
" igb-hom-[tiny|small|medium] datasets are supported.",
" igb-hom-[tiny|small|medium|large] and igb-hom datasets are supported.",
)
parser.add_argument(
"--mode",

View File

@@ -367,6 +367,7 @@ def parse_args():
"igb-hom-small",
"igb-hom-medium",
"igb-hom-large",
"igb-hom",
"reddit",
"yelp",
"flickr",

View File

@@ -343,10 +343,11 @@ def parse_args():
"igb-hom-small",
"igb-hom-medium",
"igb-hom-large",
"igb-hom",
],
help="The dataset we can use for node classification example. Currently"
" ogbn-products, ogbn-arxiv, ogbn-papers100M and"
" igb-hom-[tiny|small|medium|large] datasets are supported.",
" igb-hom-[tiny|small|medium|large] and igb-hom datasets are supported.",
)
parser.add_argument(
"--fanout",

View File

@@ -979,10 +979,11 @@ class BuiltinDataset(OnDiskDataset):
.. note::
Reverse edges are added to the original graph.
**igb-hom-[tiny|small|medium|large]**
The igb-hom-[tiny|small|medium] dataset is a homogeneous citation network,
which is designed for developers to train and evaluate GNN models with
high fidelity. See more details in `igb-hom-[tiny|small|medium|large]
**igb-hom and igb-hom-[tiny|small|medium|large]**
The igb-hom-[tiny|small|medium|large] and igb-hom dataset is a homogeneous
citation network, which is designed for developers to train and evaluate
GNN models with high fidelity. See more details in
`igb-hom-[tiny|small|medium|large]
<https://github.com/IllinoisGraphBenchmark/IGB-Datasets>`_.
.. note::
@@ -1028,6 +1029,8 @@ class BuiltinDataset(OnDiskDataset):
"igb-hom-medium-seeds",
"igb-hom-large",
"igb-hom-large-seeds",
"igb-hom",
"igb-hom-seeds",
]
_all_datasets = _datasets + _large_datasets