Unverified Commit 0b00562d authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Update gindt.py (#2212)

parent 47bb0593
"""Dataset for Graph Isomorphism Network(GIN) """Datasets used in How Powerful Are Graph Neural Networks?
(chen jun): Used for compacted graph kernel dataset in GIN (chen jun)
Data sets include: Datasets include:
MUTAG, COLLAB, IMDBBINARY, IMDBMULTI, NCI1, PROTEINS, PTC, REDDITBINARY, REDDITMULTI5K MUTAG, COLLAB, IMDBBINARY, IMDBMULTI, NCI1, PROTEINS, PTC, REDDITBINARY, REDDITMULTI5K
https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip
""" """
...@@ -17,11 +17,13 @@ from ..convert import graph as dgl_graph ...@@ -17,11 +17,13 @@ from ..convert import graph as dgl_graph
class GINDataset(DGLBuiltinDataset): class GINDataset(DGLBuiltinDataset):
"""Datasets for Graph Isomorphism Network (GIN) """Dataset Class for `How Powerful Are Graph Neural Networks? <https://arxiv.org/abs/1810.00826>`_.
Adapted from `<https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip>`_.
This is adapted from `<https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip>`_.
The dataset contains the compact format of popular graph kernel datasets.
For more graph kernel datasets, see :class:`TUDataset`. The class provides an interface for nine datasets used in the paper along with the paper-specific
settings. The datasets are ``'MUTAG'``, ``'COLLAB'``, ``'IMDBBINARY'``, ``'IMDBMULTI'``,
``'NCI1'``, ``'PROTEINS'``, ``'PTC'``, ``'REDDITBINARY'``, ``'REDDITMULTI5K'``.
Parameters Parameters
--------- ---------
......
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