supanthadey1/protein-binding-affinity

🧬 Protein Binding Affinity Predictor

Dual-head model for predicting protein-protein binding affinity (ΔG) and mutation effects (ΔΔG).

Model Performance

Architecture

• —Backbone: ESM-600M (frozen embeddings)
• —Pooling: Sliced-Wasserstein Embedding (SWE)
• —Heads: Dual-head (dG + ddG)
• —Input: Protein sequences (1153-dim = 1152 ESM + 1 mutation channel)

Usage

from huggingface_hub import hf_hub_download
import torch

# Download checkpoint
ckpt = hf_hub_download(repo_id="supanthadey1/protein-binding-affinity", filename="best_model_checkpoint.pt")
checkpoint = torch.load(ckpt, map_location='cpu')
model.load_state_dict(checkpoint['model_state_dict'])

Predictions

• —ΔG (kcal/mol): Binding free energy. More negative = stronger binding.
• —ΔΔG (kcal/mol): Mutation effect. Negative = stabilizing, Positive = destabilizing.

Training Data

Trained on multiple datasets including SKEMPI, BindingGym, PDBbind, and others.

Citation

[Citation coming soon]