Spaces:
Sleeping
Sleeping
andykr1k commited on
Commit ·
43e4bab
1
Parent(s): 20b3c7f
fixing issue
Browse files
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 |
-
|
|
|
|
|
|
|
| 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
|