mims-harvard / PINNACLE

Contextual AI models for single-cell protein biology
https://zitniklab.hms.harvard.edu/projects/PINNACLE
MIT License
72 stars 16 forks source link

Question regarding how to extract protein representation from your pretrain model #6

Open Tizzzzy opened 3 weeks ago

Tizzzzy commented 3 weeks ago

Hi author, Huge fan of your work. I am currently trying to apply your code to a downstream task. Specifically, I am working with a protein PDB file and aim to extract the latent representation of the protein using your pretrained model. I was wondering if this is possible. If it is, could you kindly show me which script I should run and which line of code store the representation of the protein? Thank you so much for your time