cogdl.models.nn.gcn
¶
Module Contents¶
Classes¶
Simple GCN layer, similar to https://arxiv.org/abs/1609.02907 |
|
The GCN model from the `”Semi-Supervised Classification with Graph Convolutional Networks” |
-
class
cogdl.models.nn.gcn.
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.gcn.
TKipfGCN
(nfeat, nhid, nclass, dropout)[source]¶ Bases:
cogdl.models.BaseModel
The GCN model from the “Semi-Supervised Classification with Graph Convolutional Networks” paper
- Args:
num_features (int) : Number of input features. num_classes (int) : Number of classes. hidden_size (int) : The dimension of node representation. dropout (float) : Dropout rate for model training.