wukevin/tcr-bert

TCR transformer model

See our full codebase and our preprint for more information.

This model is on:

• Masked language modeling (masked amino acid or MAA modeling)
• Classification across antigen labels from PIRD

If you are looking for a model trained only on MAA, please see our other model.

Example inputs:

• C A S S P V T G G I Y G Y T F (binds to NLVPMVATV CMV antigen)
• C A T S G R A G V E Q F F (binds to GILGFVFTL flu antigen)