ashu316 commited on
Commit
2e5c9f0
·
verified ·
1 Parent(s): 52ca193

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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