cogdl.models.nn.pyg_infomax
¶
Module Contents¶
Functions¶
|
-
class
cogdl.models.nn.pyg_infomax.
Encoder
(in_channels, hidden_channels)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.models.nn.pyg_infomax.
Infomax
(num_features, num_classes, hidden_size)[source]¶ Bases:
cogdl.models.BaseModel