pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix docs error in gnn api

This commit is contained in:
Zhenglong Li 2021-06-16 17:58:48 +08:00
parent 1e9786e7a8
commit 2a3e1329d8
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/dataset/engine/graphdata.py
View File

@ -204,9 +204,10 @@ class GraphData:
Get `neighbor_type` neighbors of the nodes in `node_list`.
We try to use the following example to illustrate the definition of these formats. 1 represents connected
between two nodes, and 0 represents not connected.
.. list-table:: Adjacent Matrix
:widths: 20 20 20 20 20
:header-rows: 1
:widths: 20 20 20 20 20
:header-rows: 1
* -
- 0
@ -304,9 +305,9 @@ class GraphData:
>>> nodes = graph_dataset.get_all_nodes(node_type=1)
>>> neighbors = graph_dataset.get_all_neighbors(node_list=nodes, neighbor_type=2)
>>> neighbors_coo = graph_dataset.get_all_neighbors(node_list=nodes, neighbor_type=2,
output_format=OutputFormat.COO)
>>> output_format=OutputFormat.COO)
>>> offset_table, neighbors_csr = graph_dataset.get_all_neighbors(node_list=nodes, neighbor_type=2,
output_format=OutputFormat.CSR)
>>> output_format=OutputFormat.CSR)
Raises:
TypeError: If `node_list` is not list or ndarray.