sangyx / gtrick

Bag of Tricks for Graph Neural Networks.

Hi, I'm Yunxin Sang. 👋

I'm currently working as a Ph.D. Student at McCombs School of Business, UT Austin.

I'm interested in AI for Business, exactly, I design deep learning models to solve practical business problems.

If you're interested in my projects, feel free to contact me.

gtrick's Issues

When I try to use EGNN from model.py without adding VirtualNode

Why I get this kind of error: IndexError: index out of range in self
I'm trying to compare the result before and after adding a Virtual Node.

information about the environment

Hi,
Could you plz provide information about the environment in which the code was run (https://github.com/sangyx/gtrick/blob/main/benchmark/pyg/graph_pred.py), including the version of the dependencies used? This will help me to reproduce the results you obtained. Thank you very much.

why should not use x += perturb

Hi, thanks for your work. In FLAG.py, you comment 'add perturb to x, note that do not use x += perturb' in line 65. And I want to know why should not use x += perturb.

Is the dimension of h and vx output by the 'update_node_emb' method in the 'VirtualNode' class consistent?

Hello, may I ask if the dimensions of h and vx output by the 'update_node_emb' method are the same? From the code perspective, it should be the same, but what I actually run is different. The number of lines in h is the same as the number of nodes, but the number of lines in vx is the same as "batch_size". Finally, it seems that due to the inconsistent dimensions of h and vx, the statement 'global_add_pool (x, batch)+vx' in the 'update_vn_emb' method reported an error.
The error reported is

RuntimeError: The size of tensor a (37) must match the size of tensor b (128) at non-singleton dimension 1

why

边特征提示AttributeError: type object 'SparseTensor' has no attribute 'from_edge_index'

bug

https://github.com/sangyx/gtrick/blob/main/gtrick/dgl/virtual_node.py
Is line35 super(VirtualNode).__init__() wrong?

Question about VirtualNode

I want to add a virtual node to the graph, but I don't want to connect the whole graph, just some nodes in it. How should I do this?
For example: there is a node V={0,1,2,3,4} in the graph, I want to add a node s, connect {s,0},{s,2}, how to realise this is relatively simple.