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.0.2.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.0.2-py3-none-any.whl (695.1 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: protspace-4.0.2.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.0.2.tar.gz
Algorithm Hash digest
SHA256 7a80a45245a58085968bc82091c74baeb2d7bc15825812656efb98ded8825e93
MD5 3f9ce6a7b976b7c22e04c3d412faf626
BLAKE2b-256 223ae3b16c330ca4094aeb043055ba54e95e955fc765daeb8cadbe3c8b9192f2

See more details on using hashes here.

Provenance

The following attestation bundles were made for protspace-4.0.2.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.0.2-py3-none-any.whl.

File metadata

  • Download URL: protspace-4.0.2-py3-none-any.whl
  • Upload date:
  • Size: 695.1 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.0.2-py3-none-any.whl
Algorithm Hash digest
SHA256 294b20e7c040fe594fddedf0178e6c60bdcef9ff454227a35bf6fdc92dec4b30
MD5 249bfbaa270653727d0de0e0e84ba77c
BLAKE2b-256 bc6b3ef7cfbc83798bbb4b8e95e96d1f3ce870c6101fe1dbe9f860aa7421b589

See more details on using hashes here.

Provenance

The following attestation bundles were made for protspace-4.0.2-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