nodes

Class(es) for building/connecting graphs.

class graphnet.models.graphs.nodes.nodes.NodeDefinition(*args, **kwargs)

Bases: Model

Base class for graph building.

Construct Detector.

Parameters:

• args (Any) –

• kwargs (Any) –

Return type:

object

forward(x)

Construct nodes from raw node features.

Parameters:

• x (tensor) – standardized node features with shape ´[num_pulses, d]´,

• features. (where ´d´ is the number of node) –

Returns:

a graph without edges

Return type:

graph

property nb_outputs: int

Return number of output features.

This the default, but may be overridden by specific inheriting classes.

set_number_of_inputs(node_feature_names)

Return number of inputs expected by node definition.

Parameters:

node_feature_names (List[str]) – name of each node feature column.

Return type:

None

class graphnet.models.graphs.nodes.nodes.NodesAsPulses(*args, **kwargs)

Bases: NodeDefinition

Represent each measured pulse of Cherenkov Radiation as a node.

Construct Detector.

Parameters:

• args (Any) –

• kwargs (Any) –

Return type:

object