cogdl.models.nn.pyg_stpgnn

Module Contents

Classes

stpgnn

class cogdl.models.nn.pyg_stpgnn.stpgnn(args)

Bases: cogdl.models.BaseModel

static add_args(parser)

Add model-specific arguments to the parser.

classmethod build_model_from_args(cls, args)

Build a new model instance.