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

Bases: Module

Graph SAmple and aggreGatE

forward(x, edge_index)

Forward pass through the model

predict_edges(edge_embeddings)

Predict the probability of a link between pairs of nodes