cogdl.models.nn.disengcn
¶
Module Contents¶
Classes¶
Implementation of “Disentangled Graph Convolutional Networks” <http://proceedings.mlr.press/v97/ma19a.html>. |
|
-
class
cogdl.models.nn.disengcn.
DisenGCNLayer
(in_feats, out_feats, K, iterations, tau=1.0, activation='leaky_relu')[source]¶ Bases:
torch.nn.Module
Implementation of “Disentangled Graph Convolutional Networks” <http://proceedings.mlr.press/v97/ma19a.html>.
-
class
cogdl.models.nn.disengcn.
DisenGCN
(in_feats, hidden_size, num_classes, K, iterations, tau, dropout, activation)[source]¶ Bases:
cogdl.models.BaseModel