- Clone the repository
- Create and activate a virtual environment
- Install the required dependencies:
pip install -r requirements.txt
- Download the model files from Hugging Face:
- Visit Hugging Face ESM2 Model.
- Download the necessary files and place them in the
esm_model
directory .
- Download the trained model parameters:
- If you want to use the fine-tuned model weights directly, visit Zenodo link.
- Download the trained model parameters and place them in the appropriate directory in your project.
- Train Classification Model:
bash classification.sh
- Train Regression Model:
bash regression.sh
- Run Pipeline:
To load the trained models and perform classification and regression to get candiadate seqeunces, run:
We have already screened the randomly generated sequence space, and the screening results are in the
python pipeline.py
candidate_sequences
directory.