IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Stanislaw Adaszewski 4 anni fa
parent
commit
456a037e58
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. +1
    -1
      src/decagon_pytorch/normalize.py

+ 1
- 1
src/decagon_pytorch/normalize.py Vedi File

@@ -37,7 +37,7 @@ def normalize_adjacency_matrix(adj):
def norm_adj_mat_one_node_type(adj):
adj = sp.coo_matrix(adj)
assert adj.shape[0] == adj.shape[1]:
assert adj.shape[0] == adj.shape[1]
adj_ = adj + sp.eye(adj.shape[0])
rowsum = np.array(adj_.sum(1))
degree_mat_inv_sqrt = np.power(rowsum, -0.5).flatten()


Loading…
Annulla
Salva