cleaning_module

IceCube I3InferenceModule.

Contains functionality for writing model predictions to i3 files.

class graphnet.deployment.icecube.cleaning_module.I3PulseCleanerModule(pulsemap, features, pulsemap_extractor, model_config, state_dict, model_name, *, gcd_file, threshold, discard_empty_events)[source]

Bases: I3InferenceModule

A specialized module for pulse cleaning.

It is assumed that the model provided has been trained for this.

General class for inference on I3Frames (physics).

Parameters:

  • pulsemap (str) – the pulsmap that the model is expecting as input (the one that is being cleaned).

  • features (List[str]) – the features of the pulsemap that the model is expecting.

  • pulsemap_extractor (Union[List[I3FeatureExtractor], I3FeatureExtractor]) – The extractor used to extract the pulsemap.

  • model_config (Union[ModelConfig, str]) – The ModelConfig (or path to it) that summarizes the model used for inference.

  • state_dict (str) – Path to state_dict containing the learned weights.

  • model_name (str) – The name used for the model. Will help define the named entry in the I3Frame. E.g. “dynedge”.

  • gcd_file (str) – path to associated gcd file.

  • threshold (float, default: 0.7) – the threshold for being considered a positive case. E.g., predictions >= threshold will be considered to be signal, all else noise.

  • discard_empty_events (bool, default: False) – When true, this flag will eliminate events whose cleaned pulse series are empty. Can be used to speed up processing especially for noise simulation, since it will not do any writing or further calculations.