Dataset used to train ProteinCLIP

This contains embeddings for UniProt records used to train ProteinCLIP. This includes both embeddings from protein language models and natural language emebeddings of function. All records are stored in hdf5 files with identifiers as keys (e.g., P38398) and embedding arrays as their values. All embeddings are given as 1-dimensional vectors.

The following protein language models have associated embeddings:

• ESM2, 6-layer
• ESM2, 12-layer
• ESM2, 30-layer
• ESM2, 33-layer
• ESM2, 36-layer
• ProtT5

Some of the above protein embeddings are split into sub-file indicated by the suffix "_splitN"; these files when concatenated yield the full set of proteins and are created separately due to parallel processing of embeddings.

Text embeddings are generated by OpenAI's "text-embedding-3-large" model.

We also include the raw files used to create these embeddings. Namely the ".dat.gz" file contains the archive of UniProt annotations including the function fields we parse for creating function text embeddings, and ".fasta.gz" file containing corresponding sequences.