a question about Centrality Encoding

microsoft / Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.

MIT License

2.08k stars 334 forks source link

Closed NO1excavator closed 3 years ago

NO1excavator commented 3 years ago

self.in_degree_encoder = nn.Embedding(64, hidden_dim, padding_idx=0) Why the size of the degree dictionary is 64 or 512? Thanks!

zhengsx commented 3 years ago

It should be at least (num_nodes - 1) for the possible maximum of degree in a graph.

NO1excavator commented 3 years ago

Thanks！