alpha_eps = alpha * eps
js = []
vals = []
for inode in range(len(out_degree)):
p = {inode: 0.0}
r = {}
r[inode] = alpha
q = [inode]
while len(q) > 0:
unode = q.pop()
res = r[unode] if unode in r else 0
if unode in p:
p[unode] += res
else:
p[unode] = res
r[unode] = 0
for vnode in indices[indptr[unode]:indptr[unode + 1]]:
_val = (1 - alpha) * res / out_degree[unode]
if vnode in r:
r[vnode] += _val
else:
r[vnode] = _val
res_vnode = r[vnode] if vnode in r else 0
if res_vnode >= alpha_eps * out_degree[vnode]:
if vnode not in q:
q.append(vnode)
js.append(list(p.keys()))
vals.append(list(p.values()))
return js, vals
def __repr__(self):
return "{}()".format(self.__class__.__name__)
After Change
alpha_eps = alpha * eps
js = [[0]] * len(out_degree)
vals = [[0.]] * len(out_degree)
for inode_uint in numba.prange(len(out_degree)):
inode = numba.int64(inode_uint)
p = {inode: 0.0}
r = {}
