Skip to content

Add MixTCRpred pMHC:TCR binding predictor wrapper #230

Description

@iskandr

Wrap MixTCRpred as an mhctools pMHC:TCR binding predictor (pMHC_TCR_binding).

Upstream: https://github.com/GfellerLab/MixTCRpred — a transformer predicting TCR binding to a given pMHC; one model per pMHC.

Pretrained weights: on Zenodo (record 7930623, CC-BY-4.0) — 146 per-pMHC models (~32 MB each), auto-downloaded on demand by the code; 2 are also in-repo. No retraining needed.

Framework / deps: PyTorch + PyTorch-Lightning, modern (Python 3.9–3.11) — the most current stack of the benchmark set. Colab provided.

License: code is Ludwig Institute academic / non-commercial; the weights are separately CC-BY-4.0 (freely redistributable with attribution).

Inputs / output: paired αβ CDR3 + V/J genes → binding score. The presenting pMHC is selected by which model you load (mhc_dependence effectively per-model / single_allele). Note 43/146 models are flagged "high-confidence".

Ease: Easy–moderate. Modern install; the main wrinkles are the paired-chain + V/J input requirement (richer than mhctools.TCR's CDR3-centric default — the TCR type does carry all six CDRs, but V/J genes would need to be supplied) and the one-model-per-pMHC design (map requested allele+peptide → model name).

Suggested wrapping approach: in-process via Lightning as an optional extra, or subprocess; auto-fetch the requested pMHC model from Zenodo in a setup step. Expose the model catalog so callers can see which pMHCs are covered.


Surfaced from a survey of TCREpitopeBenchmark for methods with public pretrained weights that run easily for inference.

https://claude.ai/code/session_01LZahFhBSCiehXTESCYQ7wG

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions