detector

Base detector-specific Model class(es).

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

Bases: Model

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

Construct Detector.

Parameters:
  • replace_with_identity (Optional[List[str]], default: None) – A list of feature names from the

  • identity (feature_map that should be replaced with the)

  • function.

  • args (Any)

  • kwargs (Any)

Return type:

object

abstract feature_map()[source]

List of features used/assumed by inheriting Detector objects.

Return type:

Dict[str, Callable]

forward(input_features, input_feature_names)[source]

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.