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