[Dataset] Add CIFAR10 MNIST dataset in benchmark-gnn (dmlc#6543)

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
peizhou001 · Nov 27, 2023 · 7b45c66 · 7b45c66
1 parent be8a59f
commit 7b45c66
Show file tree

Hide file tree

Showing 4 changed files with 476 additions and 2 deletions.
diff --git a/docs/source/api/python/dgl.data.rst b/docs/source/api/python/dgl.data.rst
@@ -68,9 +68,9 @@ Datasets for node classification/regression tasks
     TolokersDataset
     QuestionsDataset
     MovieLensDataset
-    PeptidesStructuralDataset
-    PeptidesFunctionalDataset
     VOCSuperpixelsDataset
+    COCOSuperpixelsDataset
+
 
 Edge Prediction Datasets
 ---------------------------------------
@@ -109,6 +109,10 @@ Datasets for graph classification/regression tasks
     FakeNewsDataset
     BA2MotifDataset
     ZINCDataset
+    PeptidesStructuralDataset
+    PeptidesFunctionalDataset
+    MNISTSuperPixelDataset
+    CIFAR10SuperPixelDataset
 
 Dataset adapters
 -------------------

diff --git a/python/dgl/data/__init__.py b/python/dgl/data/__init__.py
@@ -83,6 +83,7 @@
 except ImportError:
     pass
 from .pattern import PATTERNDataset
+from .superpixel import CIFAR10SuperPixelDataset, MNISTSuperPixelDataset
 from .wikics import WikiCSDataset
 from .yelp import YelpDataset
 from .zinc import ZINCDataset