Input Features for Chameleon and Squirrel
Closed this issue · 2 comments
Thank you for the amazing work! I am new to GNN and I am a little confused why you change the input feature of Chameleon
and Squirrel
to one hot encoding in https://github.com/LUMIA-Group/OrderedGNN/blob/main/task_node.py#L23:
data.x = torch.eye(data.x.shape[0])
Thank you so much!
Thank you for your interest! We use the setting described in Section 4.3 of [1] for Chameleon
and Squirrel
. In practice, we have found that the node features in these two datasets are less informative and can potentially mislead our gating module, which is purely data-driven. Therefore, we replace the input feature with one-hot encoding so that GNN mainly relies on the graph structure. You can refer to Table 4 and 5 in [1] to check the performance of the baseline models under this setting.
[1] Graph Neural Networks with Heterophily, AAAI 2021
Thank you for the explanation!