Skip to content

Commit

Permalink
Merge pull request graphnet-team#703 from ArturoLlorente/update_kaggl…
Browse files Browse the repository at this point in the history
…e_detector

added detector geometry attibutes to Kaggle detector class
  • Loading branch information
ArturoLlorente authored May 7, 2024
2 parents e13b779 + d10359f commit 6c88786
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 @@ -51,6 +51,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."""
feature_map = {
Expand Down

0 comments on commit 6c88786

Please sign in to comment.