/Graph_Transformer

A pytorch implementation of graph transformer for node classification

Primary LanguagePython

Graph_Transformer

A pytorch implementation of Graph Transformer for node classification.

Our implementation is based on "Do Transformers Really Perform Bad for Graph Representation" (NeurIPS'21) [paper] [github] and "Gophormer: Ego-Graph Transformer for Node Classification" [arxiv]

Run the code

Use preprocess_data.py for data preprocessing

Use main.py to train the graph transformer