ESM explainability project

This project focuses on ESM explainability

Installation

pip install fair-esm wandb umap-learn plotly

Please install pytorch according to the official instructions

Training

To generate esm embeddings please refer to the ESM repo

To run the training use the python full_train.py

Results