graphnet.models.detector.detector module

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(input_features, input_feature_names)

Pre-process graph Data features and build graph adjacency.

Return type:

Data

Parameters:

• input_features (tensor)

• input_feature_names (List[str])

property geometry_table: DataFrame

Public get method for retrieving a `Detector`s geometry table.

property string_index_name: str

Public get method for retrieving the string index column name.

property sensor_position_names: List[str]

Public get method for retrieving the xyz coordinate column names.

property sensor_index_name: str

Public get method for retrieving the sensor id column name.