|
--- |
|
base_model: westlake-repl/SaProt_35M_AF2 |
|
library_name: peft |
|
--- |
|
|
|
# Model Card for Model ID |
|
|
|
<!-- Provide a quick summary of what the model is/does. --> |
|
This model is used to predict fitness of mutant β-subunit of tryptophan synthase (TrpB). |
|
TrpB synthesizes L-tryptophan (Trp) from indole and L-serine (Ser). |
|
TrpB variant Tm9D8*, derived from the hyperthermophile Thermotoga maritima, was selected as the parent enzyme. |
|
Tm9D8* differs from wildtype TmTrpB by ten amino acid substitutions (P19G, E30G, I69V, K96L, P140L, N167D, I184F, L213P, G228S, and T292S). |
|
|
|
### Task type |
|
protein level regression |
|
|
|
### Dataset description |
|
The dataset is from [A combinatorially complete epistatic fitness landscape in an enzyme active site](https://www.biorxiv.org/content/10.1101/2024.06.23.600144v1). |
|
|
|
The dataset can also be found at [SaProtHub dataset](https://huggingface.co/datasets/SaProtHub/TrpB_fitness_landsacpe_dataset). |
|
|
|
Label means mutation fitness, here represents growth rate of E. coli strain. The maximum fitness is 1, the closer to 1, the better fitness. |
|
|
|
### Model input type |
|
Amino acid sequence |
|
|
|
### Performance |
|
test_pearson: 0.93 |
|
|
|
test_spearman: 0.38 |
|
### LoRA config |
|
lora_dropout: 0.0 |
|
|
|
lora_alpha: 16 |
|
|
|
target_modules: ["query", "key", "value", "intermediate.dense", "output.dense"] |
|
|
|
modules_to_save: ["classifier"] |
|
|
|
### Training config |
|
class: AdamW |
|
|
|
betas: (0.9, 0.98) |
|
|
|
weight_decay: 0.01 |
|
|
|
learning rate: 5e-4 |
|
|
|
epoch: 100 |
|
|
|
batch size: 100 |
|
|
|
precision: 16-mixed |