cogdl.models.nn.asgcn
¶
Module Contents¶
Classes¶
Simple GCN layer, similar to https://arxiv.org/abs/1609.02907 |
|
-
class
cogdl.models.nn.asgcn.
GraphConvolution
(in_features, out_features, bias=True)[source]¶ Bases:
torch.nn.Module
Simple GCN layer, similar to https://arxiv.org/abs/1609.02907
-
class
cogdl.models.nn.asgcn.
ASGCN
(num_features, num_classes, hidden_size, num_layers, dropout, sample_size)[source]¶ Bases:
cogdl.models.BaseModel