cogdl.models.nn.pyg_unet
¶
Module Contents¶
Classes¶
-
class
cogdl.models.nn.pyg_unet.
UNet
(num_features, num_classes, hidden_size, num_layers, dropout, num_nodes)[source]¶ Bases:
cogdl.models.BaseModel
cogdl.models.nn.pyg_unet
¶cogdl.models.nn.pyg_unet.
UNet
(num_features, num_classes, hidden_size, num_layers, dropout, num_nodes)[source]¶Bases: cogdl.models.BaseModel