detector

Base detector-specific Model class(es).

class graphnet.models.detector.detector.Detector(*args, **kwargs)

Bases: Model

Base class for all detector-specific read-ins in graphnet.

Construct Detector.

Parameters:

• args (Any) –

• kwargs (Any) –

Return type:

object

abstract feature_map()

List of features used/assumed by inheriting Detector objects.

Return type:

Dict[str, Callable]

forward(node_features, node_feature_names)

Pre-process graph Data features and build graph adjacency.

Return type:

Data

Parameters:

• node_features (tensor) –

• node_feature_names (List[str]) –