models.nn.pyg_drgat

Module Contents

Classes

DrGAT

class models.nn.pyg_drgat.DrGAT(num_features, num_classes, hidden_size, num_heads, dropout)[source]

Bases: models.BaseModel

static add_args(parser)[source]

Add model-specific arguments to the parser.

classmethod build_model_from_args(cls, args)[source]

Build a new model instance.

forward(self, x, edge_index)[source]
loss(self, data)[source]
predict(self, data)[source]