rs239 / ablm

Protein language model customized for antibodies
MIT License
62 stars 8 forks source link

Generating fixed-length embeddings from fastafile #6

Closed Aurelien-Pelissier closed 1 year ago

Aurelien-Pelissier commented 1 year ago

Hi, I generated some embeddings with the function

abmap.augment_from_fasta (fastaPath = fasta_file...)

The embeddings that I obtained are residue-level embeddings, what arguments should I pass to obtain fixed-length embeddings ?

Thanks you

lordim commented 1 year ago

Hello, so augmentation is part of the process for generating AbMAP embeddings. After you augment a sequence using augment_from_fasta, you need to take that embedding as input to the pre-trained AbMAP models in order to generate either fixed-length or residue-length embeddings. If you would like to do the entire process at once, you can check out the abmap.abmap_embed function.