relucent 0.2.0__tar.gz → 0.2.2__tar.gz

{relucent-0.2.0 → relucent-0.2.2}/PKG-INFO

@@ -1,6 +1,7 @@
 Metadata-Version: 2.4
 Name: relucent
-Version: 0.2.0
+Version: 0.2.2
+Summary: Explore the polyhedral complexes of ReLU neural networks.
 Author: Blake B. Gaines
 License-Expression: AGPL-3.0-or-later
 Requires-Dist: pandas>=2.3
@@ -18,7 +19,8 @@ Requires-Dist: torchvision ; extra == 'cli'
 Requires-Dist: pyvis>=0.3 ; extra == 'cli'
 Requires-Dist: kaleido ; extra == 'cli'
 Requires-Python: >=3.13, <3.14
-Project-URL: Homepage, https://github.com/bl-ake/relucent
+Project-URL: Repository, https://github.com/bl-ake/relucent
+Project-URL: Documentation, https://bl-ake.github.io/relucent/
 Provides-Extra: cli
 Description-Content-Type: text/markdown
 
@@ -26,7 +28,7 @@ Description-Content-Type: text/markdown
 [![Latest Release](https://img.shields.io/github/v/tag/bl-ake/relucent?label=Latest%20Release)](https://github.com/bl-ake/relucent/releases)
 
 # Relucent
-Explore polyhedral complexes associated with the activation states of ReLU neural networks
+Explore the polyhedral complexes of ReLU neural networks
 
 ## Environment Setup 
 1. Install Python 3.13
@@ -65,7 +67,7 @@ Or, get the adjacency graph of top-dimensional cells in the complex as a [Networ
 print(cplx.get_dual_graph())
 ```
 
-View the documentation for this library at (https://bl-ake.github.io/relucent/)[https://bl-ake.github.io/relucent/]
+View the documentation for this library at https://bl-ake.github.io/relucent/
 
 ## Source Code Structure
 * [model.py](src/relucent/model.py): PyTorch Module that acts as an interface between the model and the rest of the code

{relucent-0.2.0 → relucent-0.2.2}/README.md

@@ -2,7 +2,7 @@
 [![Latest Release](https://img.shields.io/github/v/tag/bl-ake/relucent?label=Latest%20Release)](https://github.com/bl-ake/relucent/releases)
 
 # Relucent
-Explore polyhedral complexes associated with the activation states of ReLU neural networks
+Explore the polyhedral complexes of ReLU neural networks
 
 ## Environment Setup 
 1. Install Python 3.13
@@ -41,7 +41,7 @@ Or, get the adjacency graph of top-dimensional cells in the complex as a [Networ
 print(cplx.get_dual_graph())
 ```
 
-View the documentation for this library at (https://bl-ake.github.io/relucent/)[https://bl-ake.github.io/relucent/]
+View the documentation for this library at https://bl-ake.github.io/relucent/
 
 ## Source Code Structure
 * [model.py](src/relucent/model.py): PyTorch Module that acts as an interface between the model and the rest of the code

{relucent-0.2.0 → relucent-0.2.2}/pyproject.toml

@@ -6,7 +6,8 @@ build-backend = "uv_build"
 
 [project]
 name = "relucent"
-version = "0.2.0"
+description = "Explore the polyhedral complexes of ReLU neural networks."
+version = "0.2.2"
 requires-python = ">= 3.13, <3.14"
 dependencies = [
     "pandas>=2.3",
@@ -35,4 +36,5 @@ cli = [
 ]
 
 [project.urls]
-Homepage = "https://github.com/bl-ake/relucent"
+Repository = "https://github.com/bl-ake/relucent"
+Documentation = "https://bl-ake.github.io/relucent/"

{relucent-0.2.0 → relucent-0.2.2}/src/relucent/poly.py

@@ -224,13 +224,19 @@ class Polyhedron:
         self._finite = self._center is not None
         return self._center, self._inradius
 
-    def get_hs(self):
+    def get_hs(self, data=None, get_all_Ab=False):
         """Get the halfspace representation of this polyhedron.
 
         Computes the halfspaces (inequality constraints) that define the polyhedron
         from all neurons in the network. The result includes constraints from
         every neuron, not just the supporting hyperplanes.
 
+        Args:
+            data: Optional input data to the network for verification. If provided,
+                checks that computed outputs match network outputs. Defaults to None.
+            get_all_Ab: If True, returns all intermediate affine maps (A, b) for
+                each layer instead of just the final halfspaces. Defaults to False.
+
         Returns:
             tuple: (halfspaces, W, b) where:
                 - halfspaces: Array of shape (n_constraints, n_dim+1) with bias terms
@@ -238,14 +244,14 @@ class Polyhedron:
                 - b: Affine transformation bias vector
         """
         if isinstance(self.bv, torch.Tensor):
-            return self.get_hs_torch()
+            return self._get_hs_torch(data, get_all_Ab)
         elif isinstance(self.bv, np.ndarray):
-            return self.get_hs_numpy()
+            return self._get_hs_numpy(data, get_all_Ab)
         else:
             raise NotImplementedError
 
     @torch.no_grad()
-    def get_hs_torch(self, data=None, get_all_Ab=False):
+    def _get_hs_torch(self, data=None, get_all_Ab=False):
         """Get halfspaces when the sign sequence is a torch.Tensor.
 
         Computes the halfspace representation using PyTorch operations.
@@ -315,7 +321,7 @@ class Polyhedron:
         return halfspaces, current_A, current_b
 
     @torch.no_grad()
-    def get_hs_numpy(self, data=None, get_all_Ab=False):
+    def _get_hs_numpy(self, data=None, get_all_Ab=False):
         """Get halfspaces when the sign sequence is a numpy array.
 
         Args: