cogdl.models.nn.mixhop

Module Contents

Classes

MixHop

class cogdl.models.nn.mixhop.MixHop(num_features, num_classes, dropout, layer1_pows, layer2_pows)

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.

forward(self, x, edge_index)

loss(self, data)

predict(self, data)