andykr1k commited on
Commit
43e4bab
·
1 Parent(s): 20b3c7f

fixing issue

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -192,7 +192,9 @@ def rebuild_model():
192
  # Get sorted lists of user and post nodes
193
  user_nodes = sorted([n for n, attr in G.nodes(data=True) if attr.get('type') == 'user'])
194
  post_nodes = sorted([n for n, attr in G.nodes(data=True) if attr.get('type') == 'post'])
195
- all_nodes = sorted(G.nodes())
 
 
196
  node2idx = {node: i for i, node in enumerate(all_nodes)}
197
 
198
  # Use identity features (one-hot) for all nodes
 
192
  # Get sorted lists of user and post nodes
193
  user_nodes = sorted([n for n, attr in G.nodes(data=True) if attr.get('type') == 'user'])
194
  post_nodes = sorted([n for n, attr in G.nodes(data=True) if attr.get('type') == 'post'])
195
+ user_nodes = sorted(n for n, attr in G.nodes(data=True) if attr.get('type') == 'user')
196
+ post_nodes = sorted(n for n, attr in G.nodes(data=True) if attr.get('type') == 'post')
197
+ all_nodes = user_nodes + post_nodes
198
  node2idx = {node: i for i, node in enumerate(all_nodes)}
199
 
200
  # Use identity features (one-hot) for all nodes