Metadata-Version: 2.4
Name: relucent
Version: 0.2.3
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
Requires-Dist: gurobipy>=12.0
Requires-Dist: networkx>=3.6
Requires-Dist: matplotlib>=3.10
Requires-Dist: numpy>=2.4
Requires-Dist: pillow>=12.1
Requires-Dist: plotly>=6.3
Requires-Dist: scikit-learn>=1.8
Requires-Dist: scipy>=1.17
Requires-Dist: tqdm>=4.67
Requires-Dist: torch>=2.13 ; extra == 'cli'
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: Repository, https://github.com/bl-ake/relucent
Project-URL: Documentation, https://bl-ake.github.io/relucent/
Provides-Extra: cli
Description-Content-Type: text/markdown

[![Usable](https://github.com/bl-ake/relucent/actions/workflows/python-package.yml/badge.svg)](https://github.com/bl-ake/relucent/actions/workflows/python-package.yml)
[![Latest Release](https://img.shields.io/github/v/tag/bl-ake/relucent?label=Latest%20Release)](https://github.com/bl-ake/relucent/releases)

# Relucent
Explore the polyhedral complexes of ReLU neural networks

## Environment Setup 
1. Install Python 3.13
2. Install [PyTorch](https://pytorch.org/get-started/locally/)
3. Run `pip install relucent`

## Getting Started
To see if the installation has been successful, try plotting the complex of a randomly initialized network in 2 dimensions like this:
```
from relucent import Complex, get_mlp_model

network = get_mlp_model(widths=[2, 10, 5, 1])
cplx = Complex(network)
cplx.bfs()
fig = cplx.plot(bound=10000)
fig.show()
```

The "NN" object returned by get_mlp_model inherits from torch.nn.Module, so you can train and manipulate it just like you're used to :)

Given some input point, you could get a minimal H-representation of the polyhedron containing it like this:
```
import numpy as np

input_point = np.random.random(network.input_shape)
p = cplx.point2poly(input_point)
print(p.halfspaces[p.shis, :])
```

You could also check the average number of faces of all polyhedrons with:
```
sum(len(p.shis) for p in cplx) / len(cplx)
```
Or, get the adjacency graph of top-dimensional cells in the complex as a [NetworkX Graph](https://networkx.org/documentation/stable/tutorial.html) with:
```
print(cplx.get_dual_graph())
```

View the documentation for this library at https://bl-ake.github.io/relucent/

## Obtaining a Gurobi License
Without a [license](https://support.gurobi.com/hc/en-us/articles/12872879801105-How-do-I-retrieve-and-set-up-a-Gurobi-license), Gurobi will only work with a limited feature set. This includes a limit on the number of decision variables in the models it can solve, which limits the size of the networks this code is able to analyze. There are multiple ways to install the software, but we recommend the following steps to those eligible for an academic license:
1. Install the [Gurobi Python library](https://pypi.org/project/gurobipy/), for example using `pip install gurobipy`
2. [Obtain a Gurobi license](https://support.gurobi.com/hc/en-us/articles/360040541251-How-do-I-obtain-a-free-academic-license) (Note: a WLS license will limit the number of concurrent sessions across multiple devices, which can result in slowdowns when using this library on different machines simultaneously.)
3. In your Conda environment, run `grbgetkey` followed by your license key
