Skip to content

GitLab

Unverified Commit a97aaa49 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

fix (#3076)

parent 7823ae2d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
python/dgl/sampling/node2vec_randomwalk.py
View file @ a97aaa49
...@@ -20,10 +20,12 @@ def node2vec_random_walk(g, nodes, p, q, walk_length, prob=None, return_eids=Fal ...@@ -20,10 +20,12 @@ def node2vec_random_walk(g, nodes, p, q, walk_length, prob=None, return_eids=Fal
Note that if a random walk stops in advance, DGL pads the trace with -1 to have the same Note that if a random walk stops in advance, DGL pads the trace with -1 to have the same
length. length.
Parameters Parameters
---------- ----------
g : DGLGraph g : DGLGraph
The graph. Must be on CPU. The graph. Must be on CPU.
Note that node2vec only support homogeneous graph. Note that node2vec only support homogeneous graph.
nodes : Tensor nodes : Tensor
Node ID tensor from which the random walk traces starts. Node ID tensor from which the random walk traces starts.
......
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