graphorge.gnn_base_model.model.gnn_epd_model

Graph Neural Network based Encoder-Process-Decoder model.

Classes

EncodeProcessDecode(torch.nn.Module)

GNN-based Encoder-Process-Decoder model.

Encoder(GraphIndependentNetwork)

GNN-based encoder.

Processor(torch_geometric.nn.MessagePassing)

GNN-based processor.

Decoder(torch.nn.Module)

FNN-based decoder.

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

Decoder(n_hidden_layers, hidden_layer_size)	GNN-based decoder.
EncodeProcessDecode(n_message_steps, ...[, ...])	GNN-based Encoder-Process-Decoder model.
Encoder(n_hidden_layers, hidden_layer_size)	GNN-based encoder.
GraphIndependentNetwork(n_hidden_layers, ...)	Graph Independent Network.
GraphInteractionNetwork(n_node_out, ...[, ...])	Graph Interaction Network.
Processor(n_message_steps, n_node_out, ...)	GNN-based processor.