graphorge.gnn_base_model.model.gnn_model

Graph Neural Network based model.

Classes

GNNEPDBaseModel(torch.nn.Module)

GNN Encoder-Processor-Decoder base model.

Functions

graph_standard_partial_fit

Perform batch fitting of standardization data scalers.

Functions

graph_standard_partial_fit(dataset, ...[, ...])

Perform batch fitting of standardization data scalers.

Classes

EncodeProcessDecode(n_message_steps, ...[, ...])	GNN-based Encoder-Process-Decoder model.
GNNEPDBaseModel(n_node_in, n_node_out, ...)	GNN Encoder-Processor-Decoder base model.
TorchStandardScaler(n_features[, mean, std, ...])	PyTorch tensor standardization data scaler.