Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/etowahadams/interprot


https://github.com/etowahadams/interprot

Last synced: 4 days ago
JSON representation

Awesome Lists containing this project

README

        

# InterProt

This repo contains tools used to interpret protein language models. `viz` contains the frontend app for visualizing SAE features. `interprot` is a python package containing tools for SAE training and interpretation.

## The visualizer

```bash
cd viz
pnpm install
pnpm run dev
```

## The python package

### Set up Docker container and run tests

```bash
docker compose build
docker compose run --rm interprot bash
pytest
```

### Running commands

Each directory under `interprot` contains a command-line tool. For example, `make_viz_files` takes in an SAE checkpoint and generates JSON files containing SAE activations used to serve the visualizer. You can run it with

```bash
cd interprot
python -m make_viz_files \
--checkpoint-files \
--output-dir
```

Or you can install this repo as a package and run it from any directory:

```bash
RUN pip install -e .
make_viz_files --checkpoint-files --output-dir
```