From 358a82ba9e3435b9a00b69aca0aa408bb7e4d64c Mon Sep 17 00:00:00 2001 From: Severin Magel Date: Sat, 30 May 2026 12:51:44 -0400 Subject: [PATCH] putting a lower bound on the charge to avoid negative arg in log10 --- src/graphnet/models/detector/icecube.py | 3 ++- src/graphnet/models/detector/nubench.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/graphnet/models/detector/icecube.py b/src/graphnet/models/detector/icecube.py index 877a0bf65..57870421e 100644 --- a/src/graphnet/models/detector/icecube.py +++ b/src/graphnet/models/detector/icecube.py @@ -39,7 +39,8 @@ def _dom_time(self, x: torch.tensor) -> torch.tensor: return (x - 1.0e04) / 3.0e4 def _charge(self, x: torch.tensor) -> torch.tensor: - return torch.log10(x) + # catch negative charges arising from input perturbation/noise + return torch.log10(1 + torch.clamp(x, min=0.0)) def _rde(self, x: torch.tensor) -> torch.tensor: return (x - 1.25) / 0.25 diff --git a/src/graphnet/models/detector/nubench.py b/src/graphnet/models/detector/nubench.py index 64844a230..18253aced 100644 --- a/src/graphnet/models/detector/nubench.py +++ b/src/graphnet/models/detector/nubench.py @@ -38,7 +38,8 @@ def _t(self, x: torch.tensor) -> torch.tensor: return x / 10e5 def _charge(self, x: torch.tensor) -> torch.tensor: - return torch.log10(1 + x) + # catch negative charges arising from input perturbation/noise + return torch.log10(1 + torch.clamp(x, min=0.0)) class FlowerS(NuBenchDetector):