Unverified Commit 7298173a authored by Tong He's avatar Tong He Committed by GitHub
Browse files

[Doc]Improve docstring for GINDataset (#2676)



* add docstring for gin

* update doc
Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
parent 174dd189
......@@ -25,6 +25,13 @@ class GINDataset(DGLBuiltinDataset):
settings. The datasets are ``'MUTAG'``, ``'COLLAB'``, ``'IMDBBINARY'``, ``'IMDBMULTI'``,
``'NCI1'``, ``'PROTEINS'``, ``'PTC'``, ``'REDDITBINARY'``, ``'REDDITMULTI5K'``.
If ``degree_as_nlabel`` is set to ``False``, then ``ndata['label']`` stores the provided node label,
otherwise ``ndata['label']`` stores the node in-degrees.
For graphs that have node attributes, ``ndata['attr']`` stores the node attributes.
For graphs that have no attribute, ``ndata['attr']`` stores the corresponding one-hot encoding
of ``ndata['label']``.
Parameters
---------
name: str
......
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