Correct spike-slab separation

sinzlab · Aug 25, 2023 · ac56731 · ac56731
1 parent 3d73583
commit ac56731
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neuralpredictors/measures/zero_inflated_losses.py b/neuralpredictors/measures/zero_inflated_losses.py
@@ -17,8 +17,8 @@ def forward(self, target, output, **kwargs):
         if loc.requires_grad:
             self.multi_clamp(loc, [0.0] * neurons_n, target.max(dim=0)[0])
 
-        zero_mask = target < loc
-        nonzero_mask = target >= loc
+        zero_mask = target <= loc
+        nonzero_mask = target > loc
 
         # spike loss
         spike_logl = torch.log(1 - q) - torch.log(loc)