Graphormer icon indicating copy to clipboard operation
Graphormer copied to clipboard

Implementation of Graphormer based on pytorch geometric

Open HelloWorldLTY opened this issue 2 years ago • 5 comments

Hi, is there a version for graphormer based on PYG? Or the encoder part of this model is purely based on pytorch? Thanks a lot.

HelloWorldLTY avatar Apr 14 '23 11:04 HelloWorldLTY

Hey there! I've implemented my own one. https://github.com/leffff/graphormer-pyg Except for several differences mentioned in the repository, and that my Floyd-Warshall algorithm is rather slow. If you are still interested in the PYG solution it would be cool, if you could give some feedback/or piece of advice about my work! I would really appreciate that

leffff avatar May 23 '23 12:05 leffff

Hey again! Now my Floyd-Warshall algorithm is optimised! https://github.com/leffff/graphormer-pyg/commit/c9b13f6542a79e4a2f4f46a5b35b3c08924565ee

leffff avatar May 28 '23 22:05 leffff

Fixed Centrality Encoding! Rewritten from cycles to PyG utils! https://github.com/leffff/graphormer-pyg/commit/e7772c834600ec2773dc2cc703e2a847260a0900

leffff avatar Jun 04 '23 11:06 leffff

Hey there! I've implemented my own one. https://github.com/leffff/graphormer-pyg Except for several differences mentioned in the repository, and that my Floyd-Warshall algorithm is rather slow. If you are still interested in the PYG solution it would be cool, if you could give some feedback/or piece of advice about my work! I would really appreciate that

Great Work! Thanks!

aofchas avatar Jun 12 '23 06:06 aofchas

The repository still has a list of optimizations to complete! Anyone is more than welcome to contribute!

leffff avatar Jun 13 '23 14:06 leffff