Release code about Contact Head prediction and evaluation

westlake-repl / SaProt

[ICLR'24 spotlight] Saprot: Protein Language Model with Structural Alphabet

MIT License

271 stars 25 forks source link

Release code about Contact Head prediction and evaluation #17

Closed CiaoHe closed 4 months ago

CiaoHe commented 4 months ago

Thx for your great job! I wonder when will release the code about Contact Head prediction and evaluation. Since the evaluation pipeline seems quite different from ESM-2 and TAPE.

LTEnjoy commented 4 months ago

Hi, thank you for you interest in our work!

We have uploaded the dataset of contact head prediction and provided scripts for training and evaluation. You could download the dataset here https://drive.google.com/drive/folders/11dNGqPYfLE3M-Mbh4U7IQpuHxJpuRr4g and run the script below:

python scripts/training.py -c config/Contact/saprot.yaml

Let me know if you have any other question!

CiaoHe commented 4 months ago

Great！I'll try this, thx for ur help!