Skip to content
Discussion options

You must be logged in to vote

I don't think there is a clear recommended way to do this. I think the Embedding approach is better than using torch.eye since it scales to larger graphs via scalability techniques such as NeighborSampler as well, and avoids the need of creating a sparse diagonal input feature matrix.

But you are right. These approaches will create a huge number of parameters and only work in transductive learning scenarios. IMO, there are two options to limit the number of parameters:

  1. Use synthetic input features instead, such as one-hot decoded node degrees (torch_geometric.transforms.OneHotDegree). This is commonly done in graph classification and works well there.
  2. Use basis decomposition to learn ini…

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@sbonner0
Comment options

Answer selected by sbonner0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants