Update expansion.py

wuch15 · Apr 13, 2022 · 9e77e2e · 9e77e2e
1 parent bed1364
commit 9e77e2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/expansion.py b/expansion.py
@@ -54,11 +54,11 @@ def graph_embedding_expansion(Otraining,usernei,alluserembs):
                 neighbor_embs += [[0.]*HIDDEN]*(NEIGHBOR_LEN-len(neighbor_embs))
             else:
                 neighbor_embs = [[0.]*HIDDEN]*NEIGHBOR_LEN
-        all_neighbor_embs.append(neighbor_embs)
+            all_neighbor_embs.append(neighbor_embs)
         all_neighbor_embs = all_neighbor_embs[:HIS_LEN]
         all_neighbor_embs += [[[0.]*HIDDEN]*HIS_LEN]*(HIS_LEN-len(all_neighbor_embs))
         user_neighbor_emb.append(all_neighbor_embs)
 
     user_neighbor_emb = np.array(user_neighbor_emb,dtype='float32')
     return user_neighbor_emb
-
+