python実装pagerank
551 ワード
import numpy as np
from functools import reduce
# ,[1,2] 1 2
edges = [[1, 2], [3, 2], [3, 4], [1, 3], [2, 3], [3, 1], [4, 1]]
feat_len = len(list(reduce(lambda x,y:set(x) | set(y),edges)))
arr = np.zeros(shape=[feat_len,feat_len])
#
link_num = np.zeros([1,feat_len])
for x,y in edges:
arr[x-1][y-1] = 1
link_num[0][y-1] += 1
arr /= link_num
# pr=0
pr = np.zeros(shape=[feat_len,1])
epoch = 100
d = 0.85
for n in range(epoch):
pr = (1-d) + arr.dot(pr) * d
print(pr.reshape(1,-1))