LBM-EPFL / PeSTo

Geometric deep learning method to predict protein binding interfaces from a protein structure.
https://pesto.epfl.ch
Other
97 stars 14 forks source link

Question regarding extracting protein representation from pre-train model #16

Open Tizzzzy opened 6 days ago

Tizzzzy commented 6 days 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