Get the neighborhood of a node #7090
-
|
Is there an easy way to get the neighborhood of a node from the edge_index on a graph with millions of nodes? |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 1 reply
-
|
Ok, edge_index[1, edge_index[0] == node_idx] works well (I found this line of code in an answer on common neighbors). Maybe adding a method "neighbors" in torch_geometric.utils could be useful to find the solution to this problem fast? I can open a pull request on it |
Beta Was this translation helpful? Give feedback.
-
|
You can also take a look at torch_geometric.utils.k_hop_subgraph |
Beta Was this translation helpful? Give feedback.
-
|
Oh ofc, thank you! |
Beta Was this translation helpful? Give feedback.
You can also take a look at torch_geometric.utils.k_hop_subgraph