cogdl.models.nn.pyg_gpt_gnn

Module Contents

Classes

GPT_GNN

Helper class that provides a standard way to create an ABC using
class cogdl.models.nn.pyg_gpt_gnn.GPT_GNN[source]

Bases: cogdl.models.supervised_model.SupervisedHomogeneousNodeClassificationModel, cogdl.models.supervised_model.SupervisedHeterogeneousNodeClassificationModel

Helper class that provides a standard way to create an ABC using inheritance.

static add_args(parser)[source]

Add task-specific arguments to the parser.

classmethod build_model_from_args(cls, args)[source]

Build a new model instance.

loss(self, data: Any) → Any[source]
predict(self, data: Any) → Any[source]
evaluate(self, data: Any, nodes: Any, targets: Any) → Any[source]
static get_trainer(taskType: Any, args) → Optional[Type[Union[GPT_GNNHomogeneousTrainer, GPT_GNNHeterogeneousTrainer]]][source]