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!
Browse Source

Fix typo.

master
Stanislaw Adaszewski 3 years ago
parent
commit
456a037e58
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/decagon_pytorch/normalize.py

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

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


Loading…
Cancel
Save