gammagl.models.GATV2Model

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

“How Attentive are 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)