dmlc--dgl
Hongzhi (Steve), Chen
704bcaf6dd
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
29 行
610 B
Python
29 行
610 B
Python
import dgl
|
|
import dgl.function as fn
|
|
import networkx as nx
|
|
import torch
|
|
|
|
N = 100
|
|
g = nx.erdos_renyi_graph(N, 0.05)
|
|
g = dgl.DGLGraph(g)
|
|
|
|
DAMP = 0.85
|
|
K = 10
|
|
|
|
|
|
def compute_pagerank(g):
|
|
g.ndata["pv"] = torch.ones(N) / N
|
|
degrees = g.out_degrees(g.nodes()).type(torch.float32)
|
|
for k in range(K):
|
|
g.ndata["pv"] = g.ndata["pv"] / degrees
|
|
g.update_all(
|
|
message_func=fn.copy_u(u="pv", out="m"),
|
|
reduce_func=fn.sum(msg="m", out="pv"),
|
|
)
|
|
g.ndata["pv"] = (1 - DAMP) / N + DAMP * g.ndata["pv"]
|
|
return g.ndata["pv"]
|
|
|
|
|
|
pv = compute_pagerank(g)
|
|
print(pv)
|