xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!18424 GNN API docs repair 

Merge pull request !18424 from lizhenglong1992/code_docs_api
This commit is contained in:
i-robot 2021-06-17 09:52:54 +08:00 committed by Gitee
parent d592d28861 2a3e1329d8
commit 1753f930d9
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.