From aed79ddbe8a96478ee1acf9e4eb8ec51bf89cbbc Mon Sep 17 00:00:00 2001 From: llorenteanaa Date: Wed, 1 May 2024 18:09:41 +0200 Subject: [PATCH] added detector geometry attibutes to Kaggle detector class. --- src/graphnet/models/detector/icecube.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/graphnet/models/detector/icecube.py b/src/graphnet/models/detector/icecube.py index c39240d7f..67a4d832e 100644 --- a/src/graphnet/models/detector/icecube.py +++ b/src/graphnet/models/detector/icecube.py @@ -50,6 +50,13 @@ def _pmt_area(self, x: torch.tensor) -> torch.tensor: class IceCubeKaggle(Detector): """`Detector` class for Kaggle Competition.""" + + geometry_table_path = os.path.join( + ICECUBE_GEOMETRY_TABLE_DIR, "icecube86.parquet" + ) + xyz = ["x", "y", "z"] + string_id_column = "string" + sensor_id_column = "sensor_id" def feature_map(self) -> Dict[str, Callable]: """Map standardization functions to each dimension of input data."""