Replies: 2 comments 2 replies
-
What is the final shape of |
Beta Was this translation helpful? Give feedback.
2 replies
-
Ideally, yes. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I have trained a graphsage model on a graph. I now want to get the embeddings for new unseen nodes.
So I pass to the model the node features and the edge index but then i get this error:
features = torch.FloatTensor(feature_list)
edges = torch.Tensor([src, dst]).type(torch. int64)
model(features, edges)
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got -2)
Any ideas?
Also, the node has edges to other unseen nodes. Should these edges be included in the data passed to the model?
Thank you for any help!
Beta Was this translation helpful? Give feedback.
All reactions