graphnet.models.detector.icecube module

IceCube-specific Detector class(es).

class graphnet.models.detector.icecube.IceCube86(*args, **kwargs)

Bases: Detector

Detector class for IceCube-86.

Construct Detector.

Parameters:

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/.local/lib/python3.10/data/geometry_tables/icecube/icecube86.parquet'

xyz = ['dom_x', 'dom_y', 'dom_z']

string_id_column = 'string'

sensor_id_column = 'sensor_id'

feature_map()

Map standardization functions to each dimension of input data.

Return type:

Dict[str, Callable]

class graphnet.models.detector.icecube.IceCubeKaggle(*args, **kwargs)

Bases: Detector

Detector class for Kaggle Competition.

Construct Detector.

Parameters:

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/.local/lib/python3.10/data/geometry_tables/icecube/icecube86.parquet'

xyz = ['x', 'y', 'z']

string_id_column = 'string'

sensor_id_column = 'sensor_id'

feature_map()

Map standardization functions to each dimension of input data.

Return type:

Dict[str, Callable]

class graphnet.models.detector.icecube.IceCubeDeepCore(*args, **kwargs)

Bases: IceCube86

Detector class for IceCube-DeepCore.

Construct Detector.

Parameters:

• args (Any)

• kwargs (Any)

Return type:

object

feature_map()

Map standardization functions to each dimension of input data.

Return type:

Dict[str, Callable]

class graphnet.models.detector.icecube.IceCubeUpgrade(*args, **kwargs)

Bases: Detector

Detector class for IceCube-Upgrade.

Construct Detector.

Parameters:

• args (Any)

• kwargs (Any)

Return type:

object

geometry_table_path = '/home/runner/.local/lib/python3.10/data/geometry_tables/icecube/icecube_upgrade.parquet'

xyz = ['dom_x', 'dom_y', 'dom_z']

string_id_column = 'string'

sensor_id_column = 'sensor_id'

feature_map()

Map standardization functions to each dimension of input data.

Return type:

Dict[str, Callable]