Graph_algorithm/OpenANE
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

not fixing numerical issue

Browse Source
This commit is contained in:
Dongzy 2018-11-18 00:01:15 +08:00
parent ae98f16ffa
commit 3a2a4acb93
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/libnrl/abrw.py
View File

@ -83,13 +83,7 @@ class ABRW(object):
P_X = row_as_probdist(X_sim)
t3 = time.time()
for i in range(P_X.shape[0]):
sum_row = P_X[i].sum()
if sum_row != 1.0: # to avoid some numerical issue...
delta = 1.0 - sum_row # delta is very very samll number say 1e-10 or even less...
P_X[i, i] = P_X[i, i] + delta # the diagnoal must be largest of the that row + delta --> almost no effect
t4 = time.time()
print('topk time: ', t2-t1, 'row normlize time: ', t3-t2, 'dealing numerical issue time: ', t4-t3)
print('topk time: ', t2-t1, 'row normlize time: ', t3-t2)
del A, X, X_compressed, X_sim
# =====================================core of our idea========================================
@ -100,8 +94,8 @@ class ABRW(object):
P = sparse.diags(alp).dot(P_A) + sparse.diags(1 - alp).dot(P_X)
print('# of single nodes for P_A: ', n - P_A.sum(axis=1).sum(), ' # of non-zero entries of P_A: ', P_A.count_nonzero())
print('# of single nodes for P_X: ', n - P_X.sum(axis=1).sum(), ' # of non-zero entries of P_X: ', np.count_nonzero(P_X))
t5 = time.time()
print('ABRW biased transition prob preprocessing time: {:.2f}s'.format(t5-t4))
t4 = time.time()
print('ABRW biased transition prob preprocessing time: {:.2f}s'.format(t4-t3))
return P
def save_embeddings(self, filename):