gammagl.models.GATModel

class GATModel(feature_dim, hidden_dim, num_class, heads, drop_rate, num_layers, name=None)

The graph attentional operator from the “Graph Attention Networks” paper.

Parameters:

• feature_dim (int) – input feature dimension.

• hidden_dim (int) – hidden dimension.

• num_class (int) – number of classes.

• heads (int) – number of attention heads.

• drop_rate (float) – dropout rate.

• name (str, optional) – model name.

forward(x, edge_index, num_nodes)