Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -114,7 +114,6 @@ def get_random_negative_node(source_node, destination_node, total_nodes):
|
|
114 |
def predict(u_id, i_id, timestamp):
|
115 |
# Before prediction
|
116 |
tgn.memory.__init_memory__() # Re-initialize memory
|
117 |
-
tgn.embeddings.__init_memory__()
|
118 |
|
119 |
# Then run prediction
|
120 |
u_array = np.array([int(u_id)]) # List of source nodes
|
|
|
114 |
def predict(u_id, i_id, timestamp):
|
115 |
# Before prediction
|
116 |
tgn.memory.__init_memory__() # Re-initialize memory
|
|
|
117 |
|
118 |
# Then run prediction
|
119 |
u_array = np.array([int(u_id)]) # List of source nodes
|