Skip to content

Commit

Permalink
added detector geometry attibutes to Kaggle detector class.
Browse files Browse the repository at this point in the history
  • Loading branch information
llorenteanaa committed May 1, 2024
1 parent ed0b94f commit aed79dd
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions src/graphnet/models/detector/icecube.py
Original file line number Diff line number Diff line change
Expand Up @@ -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."""
Expand Down

0 comments on commit aed79dd

Please sign in to comment.