Graph Transformer Encoding Networks
This is a repository of codes and experiment results for research on transformer and snowball encoding networks for graph classification, using PyTorch as backend.
Code
construct_graph.py is the code of constructing graph for PyTorch Geometric (PyG).
focal_loss.py is the code of focal loss.
scheduler.py is the code of scheduler for learning rate.
sten.py is the code of the model.