cogdl.models.nn.pyg_deepergcn
¶
Module Contents¶
Classes¶
-
class
cogdl.models.nn.pyg_deepergcn.
GENConv
(in_feat, out_feat, aggr='softmax_sg', beta=1.0, p=1.0, learn_beta=False, learn_p=False, use_msg_norm=False, learn_msg_scale=True)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.models.nn.pyg_deepergcn.
DeepGCNLayer
(in_feat, out_feat, conv, connection='res', activation='relu', dropout=0.0, checkpoint_grad=False)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.models.nn.pyg_deepergcn.
DeeperGCN
(in_feat, hidden_size, out_feat, num_layers, connection='res+', activation='relu', dropout=0.0, aggr='max', beta=1.0, p=1.0, learn_beta=False, learn_p=False, learn_msg_scale=True, use_msg_norm=False)[source]¶ Bases:
cogdl.models.BaseModel