Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "df4ad6f4acbe0924af9e722469b5d45b335b9c2e"
Unverified Commit 8c2f8ff8 authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Update citation_graph.py (#3221)

parent 45858d68
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 9 deletions
python/dgl/data/citation_graph.py
View file @ 8c2f8ff8
...@@ -386,7 +386,7 @@ class CoraGraphDataset(CitationGraphDataset): ...@@ -386,7 +386,7 @@ class CoraGraphDataset(CitationGraphDataset):
-------- --------
>>> dataset = CoraGraphDataset() >>> dataset = CoraGraphDataset()
>>> g = dataset[0] >>> g = dataset[0]
>>> num_class = g.num_classes >>> num_class = dataset.num_classes
>>> >>>
>>> # get node feature >>> # get node feature
>>> feat = g.ndata['feat'] >>> feat = g.ndata['feat']
...@@ -398,8 +398,6 @@ class CoraGraphDataset(CitationGraphDataset): ...@@ -398,8 +398,6 @@ class CoraGraphDataset(CitationGraphDataset):
>>> >>>
>>> # get labels >>> # get labels
>>> label = g.ndata['label'] >>> label = g.ndata['label']
>>>
>>> # Train, Validation and Test
""" """
def __init__(self, raw_dir=None, force_reload=False, verbose=True, reverse_edge=True): def __init__(self, raw_dir=None, force_reload=False, verbose=True, reverse_edge=True):
...@@ -531,7 +529,7 @@ class CiteseerGraphDataset(CitationGraphDataset): ...@@ -531,7 +529,7 @@ class CiteseerGraphDataset(CitationGraphDataset):
-------- --------
>>> dataset = CiteseerGraphDataset() >>> dataset = CiteseerGraphDataset()
>>> g = dataset[0] >>> g = dataset[0]
>>> num_class = g.num_classes >>> num_class = dataset.num_classes
>>> >>>
>>> # get node feature >>> # get node feature
>>> feat = g.ndata['feat'] >>> feat = g.ndata['feat']
...@@ -543,8 +541,6 @@ class CiteseerGraphDataset(CitationGraphDataset): ...@@ -543,8 +541,6 @@ class CiteseerGraphDataset(CitationGraphDataset):
>>> >>>
>>> # get labels >>> # get labels
>>> label = g.ndata['label'] >>> label = g.ndata['label']
>>>
>>> # Train, Validation and Test
""" """
def __init__(self, raw_dir=None, force_reload=False, verbose=True, reverse_edge=True): def __init__(self, raw_dir=None, force_reload=False, verbose=True, reverse_edge=True):
...@@ -673,7 +669,7 @@ class PubmedGraphDataset(CitationGraphDataset): ...@@ -673,7 +669,7 @@ class PubmedGraphDataset(CitationGraphDataset):
-------- --------
>>> dataset = PubmedGraphDataset() >>> dataset = PubmedGraphDataset()
>>> g = dataset[0] >>> g = dataset[0]
>>> num_class = g.num_of_class >>> num_class = dataset.num_of_class
>>> >>>
>>> # get node feature >>> # get node feature
>>> feat = g.ndata['feat'] >>> feat = g.ndata['feat']
...@@ -685,8 +681,6 @@ class PubmedGraphDataset(CitationGraphDataset): ...@@ -685,8 +681,6 @@ class PubmedGraphDataset(CitationGraphDataset):
>>> >>>
>>> # get labels >>> # get labels
>>> label = g.ndata['label'] >>> label = g.ndata['label']
>>>
>>> # Train, Validation and Test
""" """
def __init__(self, raw_dir=None, force_reload=False, verbose=True, reverse_edge=True): def __init__(self, raw_dir=None, force_reload=False, verbose=True, reverse_edge=True):
......
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