TCR-specificity-prediction-with-ESMv2

A transfer learning method of TCR specificity prediction based on ESMv2 protein language model

To replicate the results please follow the instructions below:

1. If you wish to replicate the dataset generation step please install a conda environment from requirments_dataset_generation_env.txt and follow the dataset_generation.ipynb at main/notebooks
2. If you wish to replicate the training step please install a conda environment from requirmnets_training.txt and follow the training.ipynb at main/notebooks
3. If you wish to run tests on the pretrained model please use the environment from step 2 and load the model from weights located at main/weights following the tests.ipynb at main/notebooks