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.3.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.3.0-py3-none-any.whl (705.4 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: protspace-4.3.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.3.0.tar.gz
Algorithm Hash digest
SHA256 64dd44e234a22020a80cd287f9e6cf26245b53bf0ac5d54bd17f31076390e077
MD5 2e2ef0ae91df13916978ae4be96af66c
BLAKE2b-256 c269e854e449dbc6290deac14d4ea7a7831369c4273196bc8fac32ec1d2f468f

See more details on using hashes here.

Provenance

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

File metadata

  • Download URL: protspace-4.3.0-py3-none-any.whl
  • Upload date:
  • Size: 705.4 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.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 4d7cf7f042327868eb67cb9d77b5d7523c58213cb7b97c1217357bb830dd9ce3
MD5 68baf4b752b1a555b70a1e28bbdc7af3
BLAKE2b-256 921021dd6da9dde596f9ae95765cd412bd81694ec81f2240ae8692307709e394

See more details on using hashes here.

Provenance

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