mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-03 19:34:33 +08:00
[Graphbolt][Dataset] Add igb-hom dataset (#7781)
Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com>
This commit is contained in:
@@ -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")
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -367,6 +367,7 @@ def parse_args():
|
||||
"igb-hom-small",
|
||||
"igb-hom-medium",
|
||||
"igb-hom-large",
|
||||
"igb-hom",
|
||||
"reddit",
|
||||
"yelp",
|
||||
"flickr",
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user