class GAT(*args: Any, **kwargs: Any)

Bases: Module

Graph Attention Network with customizable layers and activation

forward(x, edge_index)

Forward pass through the model

predict_edges(edge_embeddings)

Predict the probability of a link between pairs of nodes