cogdl.layers.strategies_layers
¶
Module Contents¶
Classes¶
-
class
cogdl.layers.strategies_layers.
GINConv
(hidden_size, input_layer=None, edge_emb=None, edge_encode=None, pooling='sum', feature_concat=False)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.layers.strategies_layers.
GNN
(num_layers, hidden_size, JK='last', dropout=0.5, input_layer=None, edge_encode=None, edge_emb=None, num_atom_type=None, num_chirality_tag=None, concat=False)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.layers.strategies_layers.
GNNPred
(num_layers, hidden_size, num_tasks, JK='last', dropout=0, graph_pooling='mean', input_layer=None, edge_encode=None, edge_emb=None, num_atom_type=None, num_chirality_tag=None, concat=True)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.layers.strategies_layers.
Pretrainer
(args, transform=None)[source]¶ Bases:
torch.nn.Module