Dense and Sparse Implementation of GAT written by PyTorch.
Use IF_SPARSE
to switch the dense and sparse implementations.
node_classification.py
is an example of using GAT on Node Classification Task.
link_prediction.py
is an example of using GAT on Link Prediction Task.
smyconan / pytorchgat Goto Github PK
View Code? Open in Web Editor NEWDense and Sparse Implementation of GAT written by PyTorch