Skip to main content

A visualisation tool for protein embeddings from pLMs

Project description

ProtSpace

PyPI version Python 3.10+ License: GPL v3 Downloads DOI

ProtSpace is a visualization tool for exploring protein embeddings or similarity matrices. It projects high-dimensional protein language model data into 2D space, color-codes proteins by biological annotations, and exports publication-ready figures.

  • Multiple projections: PCA, UMAP, t-SNE, MDS, PaCMAP, LocalMAP
  • Automatic annotations: UniProt, InterPro, and Taxonomy
  • Structure viewer: Integrated protein structure visualization
  • Export: PNG, PDF, SVG, HTML

🌐 Try Online

ProtSpace Web: Fast 2D explorer optimized for large datasets — drag & drop .parquetbundle files (source)

🚀 Google Colab Notebooks

Note: Use Chrome or Firefox for best experience.

  1. Generate Protein Embeddings: Open Embeddings In Colab

  2. Prepare ProtSpace Bundle: Open Preparation In Colab

📦 Installation

pip install protspace

🎯 Quick Start

1. Prepare data

# From HDF5 embeddings
protspace prepare -i embeddings.h5 -m pca2,umap2 -o output

# From FASTA (auto-embeds via Biocentral API)
protspace prepare -i sequences.fasta -e prot_t5 -m pca2 -o output

# Multi-model comparison (12 pLMs supported)
protspace prepare -i sequences.fasta -e prot_t5,esm2_650m,ankh_base -m pca2,umap2 -o output

2. Explore results

Upload the generated .parquetbundle file at protspace.app/explore.

3. Power-user workflow (individual steps)

protspace embed -i sequences.fasta -e prot_t5 -e esm2_3b -o embeddings/
protspace project -i embeddings/prot_t5.h5 -i embeddings/esm2_3b.h5 -m pca2,umap2 -o projections/
protspace annotate -i embeddings/prot_t5.h5 -a default -o annotations.parquet
protspace bundle -p projections/ -a annotations.parquet -o output.parquetbundle

📊 Example Output

2D Example

✨ Annotations

Use -a to color-code proteins by UniProt, InterPro, or Taxonomy annotations. Groups (default, all, uniprot, interpro, taxonomy) and individual names can be mixed freely. If -a is omitted, the default group is used.

protspace prepare -i data.h5 -m pca2                              # default annotations
protspace prepare -i data.h5 -a default,interpro,kingdom -m pca2  # mix groups + individual

📖 Documentation

📝 Citation

Senoner T, Olenyi T, Heinzinger M, Spannagl A, Bouras G, Rost B, Koludarov I. ProtSpace: A Tool for Visualizing Protein Space. Journal of Molecular Biology, 168940, 2025. doi:10.1016/j.jmb.2025.168940

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

protspace-4.2.0.tar.gz (10.6 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

protspace-4.2.0-py3-none-any.whl (694.8 kB view details)

Uploaded Python 3

File details

Details for the file protspace-4.2.0.tar.gz.

File metadata

  • Download URL: protspace-4.2.0.tar.gz
  • Upload date:
  • Size: 10.6 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for protspace-4.2.0.tar.gz
Algorithm Hash digest
SHA256 484e99f458fac8254750509b2a874b865e84838d971d533366e3c64894cdb86f
MD5 4a61dbaffcc94d8c11cb1b24a8682128
BLAKE2b-256 67e273c3b353c385ba721e37706fceab99b362c2b4a2203237a57e9e72a08471

See more details on using hashes here.

Provenance

The following attestation bundles were made for protspace-4.2.0.tar.gz:

Publisher: publish.yml on tsenoner/protspace

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file protspace-4.2.0-py3-none-any.whl.

File metadata

  • Download URL: protspace-4.2.0-py3-none-any.whl
  • Upload date:
  • Size: 694.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for protspace-4.2.0-py3-none-any.whl
Algorithm Hash digest
SHA256 ee996a4d810b4e2127aa030cb48ed671695d9dbb01fb4e73d06f4c704040f465
MD5 438550507ca47ea126f88f3c51ca3733
BLAKE2b-256 da10cbf3e4c7a982bc22db49983d89914a86c09fd4cce06285fa304691d52aa3

See more details on using hashes here.

Provenance

The following attestation bundles were made for protspace-4.2.0-py3-none-any.whl:

Publisher: publish.yml on tsenoner/protspace

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page