graphorge.gnn_base_model.model.gnn_architectures

Graph Neural Networks architectures.

Classes

GraphIndependentNetwork(torch.nn.Module)

Graph Independent Network.

GraphInteractionNetwork(torch_geometric.nn.MessagePassing)

Graph Interaction Network.

Functions

build_fnn

Build multilayer feed-forward neural network.

Functions

build_fnn(input_size, output_size[, ...])

Build multilayer feed-forward neural network.

build_rnn(input_size, hidden_layer_sizes, ...)

Build multilayer recurrent neural network.

Classes

GraphIndependentNetwork(n_hidden_layers, ...)

Graph Independent Network.

GraphInteractionNetwork(n_node_out, ...[, ...])

Graph Interaction Network.

TorchRNNWrapper(*args, **kwargs)

TorchRNNWrapper.