GNNBackBone Class Reference

Implements a GNN architecture, according to hyperparameters specified in the input config. More...

Inheritance diagram for GNNBackBone:

Public Member Functions
	__init__ (self, config, input_dim=None)
torch.Tensor	forward (self, torch.Tensor x, torch.Tensor edge_index, edge_weight=None)
Public Member Functions inherited from GNNConfigParser
None	parse (self, dict config)
	Parse the configuration for the GNN model.

Public Attributes
	input_dim = input_dim
	num_layers
Public Attributes inherited from GNNConfigParser
	config = None
	input_dim = None
	output_dim = None
	num_hops = None
	num_layers = None
	latent_size = None

Detailed Description

Implements a GNN architecture, according to hyperparameters specified in the input config.

Definition at line 35 of file gnn_backbone.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			config,
			input_dim = None )

Reimplemented from GNNConfigParser.

Definition at line 37 of file gnn_backbone.py.

Member Function Documentation

forward()

torch.Tensor forward	(		self,
		torch.Tensor	x,
		torch.Tensor	edge_index,
			edge_weight = None )

Definition at line 64 of file gnn_backbone.py.

Member Data Documentation

input_dim

input_dim = input_dim

Definition at line 43 of file gnn_backbone.py.

num_layers

num_layers

Definition at line 54 of file gnn_backbone.py.