search

rs239 / ablm

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

Load .p embeddings generated from fastafile #4

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...)

This created a .p files for each sequence. But I did not manage to read them. torch.load() gave me error. Could you clarify how to load these generated files?

Thanks you

lordim commented 1 year ago

Hello, the .p files generated for each sequence are pickle files. So, you would need to use pickle.load() instead of torch.load().

Aurelien-Pelissier commented 1 year ago

Thanks, it worked, I'm closing the issue